Merge pull request #1 from EmraldT95/patch-1

Fixed module import issue.
automl · Oct 16, 2024 · 5a9092e · 5a9092e
2 parents ac599c3 + d3e2e51
commit 5a9092e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_cpr/wrapper.py b/pytorch_cpr/wrapper.py
@@ -1,7 +1,7 @@
 import inspect
 
-from pytorch_cpr.optim_cpr import CPR
-from pytorch_cpr.group_parameter import cpr_group_named_parameters
+from .optim_cpr import CPR
+from .group_parameter import cpr_group_named_parameters
 
 
 def apply_CPR(model, optimizer_cls, kappa_init_param, kappa_init_method='warm_start', reg_function='l2',