whitespace

autonlab · Dec 28, 2020 · a69418e · a69418e
1 parent 9ea6c2c
commit a69418e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/dsm/__init__.py b/dsm/__init__.py
@@ -183,4 +183,4 @@
 
 from dsm.dsm_api import DeepSurvivalMachines
 from dsm.dsm_api import DeepConvolutionalSurvivalMachines
-from dsm.dsm_api import DeepRecurrentSurvivalMachines
+from dsm.dsm_api import DeepRecurrentSurvivalMachines
diff --git a/dsm/dsm_api.py b/dsm/dsm_api.py
@@ -28,7 +28,7 @@
 """
 
 from dsm.dsm_torch import DeepSurvivalMachinesTorch
-from dsm.dsm_torch import DeepRecurrentSurvivalMachinesTorch 
+from dsm.dsm_torch import DeepRecurrentSurvivalMachinesTorch
 from dsm.dsm_torch import DeepConvolutionalSurvivalMachinesTorch
 from dsm.losses import predict_cdf
 import dsm.losses as losses
@@ -353,4 +353,3 @@ def _gen_torch_model(self, inputdim, optimizer, risks):
                                               optimizer=optimizer,
                                               typ=self.typ,
                                               risks=risks)
-