commiting the changes I described earlier in #176

navis-org · Dec 16, 2024 · 628c0c1 · 628c0c1
1 parent f5b45ab
commit 628c0c1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/navis/config.py b/navis/config.py
@@ -59,6 +59,12 @@ def get_logger(name: str):
         return logging.getLogger(name)
     return logger
 
+# Set up numpy number representation, see NEP51
+# Once numpy<=2 is dropped from requirements, the doctest comparissons
+# should become `np.float64(1.074)` instead of `1.074`
+if os.environ.get('NAVIS_TEST_ENV', '').lower() == 'true':
+    import numpy as np
+    np.set_printoptions(legacy="1.25")
 
 # Default settings for progress bars
 pbar_hide = False

diff --git a/navis/morpho/mmetrics.py b/navis/morpho/mmetrics.py
@@ -15,7 +15,6 @@
 """This module contains functions to analyse and manipulate neuron morphology."""
 
 import math
-import os
 import itertools
 import scipy
 import warnings
@@ -31,11 +30,6 @@
 # Set up logging
 logger = config.get_logger(__name__)
 
-# Set up numpy number representation, see NEP51
-# Once numpy<=2 is dropped from requirements, the doctest comparissons
-# should become `np.float64(1.074)` instead of `1.074`
-if os.environ.get('NAVIS_TEST_ENV', '').lower() == 'true':
-    np.set_printoptions(legacy="1.25")
 
 __all__ = sorted(
     [