Removed some imports that load all of SciPy

Author: davenquinn

py-packages/engine/corelle/engine/cache.py

@@ -2,10 +2,12 @@
 Mapped database models. These would ideally be in the database module, but
 they need to be here to avoid initialization issues.
 """
+
 from sqlalchemy.sql import select
 
 from rich.progress import Progress
 from sqlalchemy.dialects.postgresql import insert
+from sqlalchemy.sql import text
 import numpy as N
 
 from corelle.math import quaternion_to_euler
@@ -47,16 +49,16 @@ def build_rotation_caches():
 def build_rotation_cache(model, time_step=1):
     # Get the time span of the model
 
-    min_age = model["min_age"] or 0
-    max_age = model["max_age"] or 1000
+    min_age = model.min_age or 0
+    max_age = model.max_age or 1000
 
     # Get model steps every 1 Myr
     t_steps = list(range(int(min_age), int(max_age) + 1, time_step))
 
     session = db.session()
 
     session.execute(
-        "DELETE FROM corelle.rotation_cache WHERE model_id = :model_id",
+        text("DELETE FROM corelle.rotation_cache WHERE model_id = :model_id"),
         dict(model_id=model.id),
     )
     session.commit()

py-packages/engine/corelle/engine/cli.py

@@ -1,11 +1,9 @@
 import warnings
 
-import numpy as N
 from IPython import embed
 from os import environ
 import click
 from click import group, argument, option, echo
-from corelle.math import quaternion_to_euler
 from macrostrat.utils import working_directory
 from pathlib import Path
 
@@ -104,6 +102,8 @@ def rotate(model, plate, time, verbose=False):
     """
     Rotate a plate to a time
     """
+    import numpy as N
+    from corelle.math import quaternion_to_euler
     from .rotate import get_rotation
 
     q = get_rotation(model, plate, time, verbose=verbose)
@@ -124,6 +124,7 @@ def rotate_all(model, time, verbose=False):
     """
     Rotate all plates in a model to a time
     """
+    import numpy as N
     from .rotate import get_all_rotations
 
     for plate_id, q in get_all_rotations(model, time, verbose=verbose):