Fix test_loop and relax python version (#7)

* fix test_loop * relax python version * restrict orbax version
cgarciae · Mar 16, 2023 · ce3a24e · ce3a24e
1 parent f83c7f9
commit ce3a24e
Show file tree

Hide file tree

Showing 3 changed files with 507 additions and 862 deletions.
diff --git a/ciclo/loops/common.py b/ciclo/loops/common.py
@@ -194,6 +194,8 @@ def test_loop(
     if tasks is None:
         tasks = {}
 
+    # create a copy of tasks to avoid modifying the original
+    tasks = tasks.copy()
     additionl_tasks: Dict[ScheduleLike, CallbackOrList] = {}
     named_tasks: Dict[str, CallbackOrList] = {}
     for schedule in list(tasks.keys()):