Update R wrapper logic

microsoft · Jun 16, 2021 · 0fe8b28 · 0fe8b28
1 parent 64582e9
commit 0fe8b28
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/qlib/workflow/__init__.py b/qlib/workflow/__init__.py
@@ -533,9 +533,11 @@ class RecorderWrapper(Wrapper):
 
     def register(self, provider):
         if self._provider is not None:
-            raise RecorderInitializationError(
-                "Please don't reinitialize Qlib if QlibRecorder is already acivated. Otherwise, the experiment stored location will be modified."
-            )
+            expm = getattr(self._provider, "exp_manager")
+            if expm.active_experiment is not None:
+                raise RecorderInitializationError(
+                    "Please don't reinitialize Qlib if QlibRecorder is already acivated. Otherwise, the experiment stored location will be modified."
+                )
         self._provider = provider