Added check for relative package imports

TDAmeritrade · Jan 21, 2025 · b6d9bce · b6d9bce · NimaSarajpoor · Jan 21, 2025
1 parent 70e4e70
commit b6d9bce
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/stumpy/core.py b/stumpy/core.py
@@ -3689,7 +3689,7 @@ def check_ignore_trivial(T_A, T_B, ignore_trivial):
 
     Notes
     -----
-    These warnings may be supresse by using a context manager
+    These warnings may be supressed by using a context manager
     ```
     import stumpy
     import numpy as np

diff --git a/test.sh b/test.sh
@@ -100,6 +100,16 @@ check_fastmath()
     check_errs $?
 }
 
+check_pkg_imports()
+{
+    echo "Checking Package Imports"
+    if [[ `grep '^from stumpy' stumpy/*py | wc -l` -gt "0" ]]; then
+        grep '^from stumpy' stumpy/*py
+        echo 'Error: please change "from stumpy" to "from ." '
+        exit 1
+    fi
+}
+
 check_naive()
 {
     # Check if there are any naive implementations not at start of test file
@@ -337,6 +347,7 @@ check_isort
 check_flake
 check_docstrings
 check_print
+check_pkg_imports
 check_naive
 check_ray