decrease predict iters to 5

BasisResearch · Oct 8, 2024 · dbddefd · dbddefd
1 parent be72dbf
commit dbddefd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cities/deployment/tracts_minneapolis/predict.py b/cities/deployment/tracts_minneapolis/predict.py
@@ -19,6 +19,10 @@
 
 load_dotenv()
 
+local_user = os.getenv("USER")
+if local_user == "rafal":
+    load_dotenv(os.path.expanduser("~/.env_pw"))
+
 
 class TractsModelPredictor:
     kwargs = {
@@ -228,7 +232,7 @@ def predict_cumulative(self, conn, intervention):
         predictor = TractsModelPredictor(conn)
         start = time.time()
 
-        for iter in range(100): # added for time testing
+        for iter in range(5): # added for time testing
             result = predictor.predict_cumulative(
                 conn,
                 intervention={