Update test_dsm.py

autonlab · Jan 30, 2022 · 92be790 · 92be790
1 parent 096ecc6
commit 92be790
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_dsm.py b/tests/test_dsm.py
@@ -4,9 +4,9 @@
 """
 import unittest
 
-from dsm import DeepSurvivalMachines
-from dsm.dsm_torch import DeepSurvivalMachinesTorch
-from dsm import datasets
+from auton_survival.models.dsm import DeepSurvivalMachines
+from auton_survival.models.dsm.dsm_torch import DeepSurvivalMachinesTorch
+from auton_survival.models.dsm import datasets
 
 import numpy as np