Merge f6fcf37 into cda0495

rht · web-flow · commit 642324c51c2d · 2021-12-20T13:45:45.000Z
diff --git a/examples/bank_reserves/batch_run.py b/examples/bank_reserves/batch_run.py
@@ -186,15 +186,6 @@ def run_model(self):
     "reserve_percent": 5,
 }
 
-"""br = BatchRunner(
-    BankReservesModel,
-    br_params,
-    iterations=2,
-    max_steps=1000,
-    nr_processes=None,
-    # model_reporters={"Data Collector": lambda m: m.datacollector},
-)"""
-
 if __name__ == "__main__":
     data = batch_run(
         BankReservesModel,
@@ -204,8 +195,20 @@ def run_model(self):
     )
     br_df = pd.DataFrame(data)
     br_df.to_csv("BankReservesModel_Data.csv")
-    # br.run_all()
+
+    # The commented out code below is the equivalent code as above, but done
+    # via the legacy BatchRunner class. This is a good example to look at if
+    # you want to migrate your code to use `batch_run()` from `BatchRunner`.
     """
+    br = BatchRunner(
+        BankReservesModel,
+        br_params,
+        iterations=2,
+        max_steps=1000,
+        nr_processes=None,
+        # model_reporters={"Data Collector": lambda m: m.datacollector},
+    )
+    br.run_all()
     br_df = br.get_model_vars_dataframe()
     br_step_data = pd.DataFrame()
     for i in range(len(br_df["Data Collector"])):
