Remove research imports

RedTachyon · Jan 16, 2024 · 4704384 · 4704384
1 parent 37fb9f3
commit 4704384
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/coltra/__init__.py b/coltra/__init__.py
@@ -6,5 +6,5 @@
 from coltra.buffers import Action, Observation
 from coltra import utils
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 VERSION = __version__
diff --git a/coltra/research/__init__.py b/coltra/research/__init__.py
@@ -1 +0,0 @@
-from .policy_fusion import JointModel

diff --git a/tests/research/test_fusion.py b/tests/research/test_fusion.py
@@ -12,7 +12,7 @@
 from coltra.models import MLPModel
 import pytest
 
-from coltra.research.policy_fusion import JointModel
+from coltra.research.policy_fusion.policy_fusion import JointModel
 
 
 def assert_models_equal(model1: nn.Module, model2: nn.Module):