Updated model.py so it runs setup.py

scaleoutsystems · Oct 28, 2024 · 47f41a6 · 47f41a6
1 parent 60e0071
commit 47f41a6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/client/model.py b/client/model.py
@@ -3,6 +3,7 @@
 import torch
 import collections
 import tempfile
+import subprocess
 
 HELPER_MODULE = "numpyhelper"
 helper = get_helper(HELPER_MODULE)
@@ -51,5 +52,10 @@ def init_seed(out_path):
     model = compile_model()
     save_parameters(model, out_path)
 
+def run_setup():
+    """Run setup.py before the main code execution."""
+    subprocess.run(["python3", "setup.py"], check=True)
+
 if __name__ == "__main__":
+    run_setup()
     init_seed('../seed.npz')