update

esa · Nov 25, 2024 · ce77b2a · ce77b2a
1 parent b91bc7c
commit ce77b2a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dsgp4/__init__.py b/dsgp4/__init__.py
@@ -10,6 +10,6 @@
 from .sgp4_batched import sgp4_batched
 from .util import propagate, initialize_tle, propagate_batch
 from .plot import plot_orbit, plot_tles
+from .mldsgp4 import mldsgp4
 from . import tle
 from .tle import TLE
-from .mldsgp4 import mldsgp4
diff --git a/dsgp4/mldsgp4.py b/dsgp4/mldsgp4.py
@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 
-from . import initialize_tle, propagate, propagate_batch
+from .util import initialize_tle, propagate, propagate_batch
 from torch.nn.parameter import Parameter
 
 class mldsgp4(nn.Module):