add AMI id to instance info on builds

apache · Feb 21, 2020 · b21c2e9 · b21c2e9
1 parent 88509d1
commit b21c2e9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ci/build_windows.py b/ci/build_windows.py
@@ -35,7 +35,7 @@
 from enum import Enum
 from subprocess import check_call
 
-from util import *
+from util import ec2_instance_info
 
 KNOWN_VCVARS = {
     'VS 2015': r'C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\x86_amd64\vcvarsx86_amd64.bat',
@@ -221,6 +221,9 @@ def main():
     logging.getLogger().setLevel(logging.INFO)
     logging.basicConfig(format='%(asctime)-15s %(message)s')
     logging.info("MXNet Windows build helper")
+    instance_info = ec2_instance_info()
+    if instance_info:
+        logging.info("EC2: %s", instance_info)
 
     parser = argparse.ArgumentParser()
     parser.add_argument("-o", "--output",