fix: resume experiment will reset to running (#185)

Signed-off-by: kerthcet <kerthcet@gmail.com>

Author: kerthcet

alphatrion/experiment/base.py

@@ -217,6 +217,11 @@ def _start(
         # to avoid confusion.
         if exp_obj and exp_obj.status != Status.COMPLETED:
             self._id = exp_obj.uuid
+            # reset to running status.
+            self._runtime._metadb.update_experiment(
+                experiment_id=self._id,
+                status=Status.RUNNING,
+            )
         elif exp_obj and exp_obj.status == Status.COMPLETED:
             raise RuntimeError(
                 f"Experiment with name '{name}' already exists and is completed. \

tests/unit/experiment/test_experimant.py

@@ -178,6 +178,16 @@ async def test_experiment_with_done_with_cancel():
 
     assert global_runtime().metadb.get_run(run_id=run_id).status == Status.CANCELLED
 
+    async with CraftExperiment.start(name="first-experiment") as exp:
+        run = exp.run(lambda: asyncio.sleep(2))
+        exp_obj = global_runtime().metadb.get_experiment(experiment_id=exp.id)
+        # will be reset to running
+        assert exp_obj.status == Status.RUNNING
+
+    # finally should be completed after context exit
+    exp_obj = global_runtime().metadb.get_experiment(experiment_id=exp.id)
+    assert exp_obj.status == Status.COMPLETED
+
 
 @pytest.mark.asyncio
 async def test_experiment_with_wait():