Fix neural argv

EnzymeAD · Nov 8, 2024 · 9876c72 · 9876c72
1 parent d249cc6
commit 9876c72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/neuralgcm_test.py b/test/neuralgcm_test.py
@@ -180,7 +180,7 @@ def run_on_fn(self, fn, steps=1):
         ).timeit(steps)
 
 
-def main():
+def main(argv):
     c = NeuralGCM()
     c.setUp()
     c.test()