Update setup.py

mlcommons · Dec 7, 2024 · 9d7411a · 9d7411a
1 parent b488f91
commit 9d7411a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,4 @@
-# Build a whl file for cm4mlperf-inference
+# Build a whl file for mlperf-automations
 
 from setuptools import setup
 from setuptools._distutils.dist import Distribution
@@ -147,10 +147,10 @@ def custom_function(self):
                           'all': True})
         r = cmind.access({'action': 'pull',
                           'automation': 'repo',
-                          'artifact': 'mlcommons@cm4mlops',
-                          'branch': 'mlperf-inference',
+                          'artifact': 'mlcommons@mlperf-automations',
+                          'branch': 'main',
                           'checkout': commit_hash})
-        # r = cmind.access({'action':'pull', 'automation':'repo', 'artifact':'mlcommons@cm4mlops', 'checkout': commit_hash})
+        # r = cmind.access({'action':'pull', 'automation':'repo', 'artifact':'mlcommons@mlperf-automations', 'checkout': commit_hash})
         print(r)
         if r['return'] > 0:
             return r['return']