Fixed it by removing the model from each test case after execution

Author: jmafoster1

causal_testing/main.py

@@ -361,6 +361,9 @@ def run_tests_in_batches(self, batch_size: int = 100, silent: bool = False) -> L
                 for test_case in current_batch:
                     try:
                         batch_results.append(test_case.execute_test())
+                        # Need to remove the model so we don't take up all the memory
+                        # Would be good to profile the execute_test() method a bit further so we don't need to do this
+                        test_case.estimator.model = None
                     # pylint: disable=broad-exception-caught
                     except Exception as e:
                         if not silent: