fixed invalid import statement

BaderTim · Jan 28, 2024 · 2da21b9 · 2da21b9
1 parent ae86574
commit 2da21b9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/baselines/model/tests/test_icp.py b/baselines/model/tests/test_icp.py
@@ -21,8 +21,8 @@
 import numpy as np
 import time
 import torch
-from baselines.model.ICP.ICP import ICP
-from baselines.model.ICP.module import icp
+from model.ICP.ICP import ICP
+from model.ICP.module import icp
 
 
 def test_icp_module():

diff --git a/baselines/model/tests/test_random.py b/baselines/model/tests/test_random.py
@@ -1,6 +1,6 @@
 import torch
 
-from baselines.model.RANDOM.RANDOM import RANDOM
+from model.RANDOM.RANDOM import RANDOM
 
 def test_random_forward_pass():
     K = 2

diff --git a/baselines/model/tests/test_zero.py b/baselines/model/tests/test_zero.py
@@ -1,6 +1,6 @@
 import torch
 
-from baselines.model.ZERO.ZERO import ZERO
+from model.ZERO.ZERO import ZERO
 
 def test_zero_forward_pass():
     K = 2