revise fastmath script and add support for reading arg from command line

Author: NimaSarajpoor

check_fastmath.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python
+
+import argparse
+import importlib
+
+from stumpy.cache import get_njit_funcs
+
+
+def check_fastmath():
+    """
+    Check if all njit functions have the `fastmath` flag set
+
+    Parameters
+    ----------
+    None
+
+    Returns
+    -------
+    None
+    """
+    missing_fastmath = []  # list of njit functions with missing fastmath flags
+    for module_name, func_name in get_njit_funcs():
+        module = importlib.import_module(f".{module_name}", package="stumpy")
+        func = getattr(module, func_name)
+        if "fastmath" not in func.targetoptions.keys():
+            missing_fastmath.append(f"{module_name}.{func_name}")
+
+    if len(missing_fastmath) > 0:
+        msg = "Found one or more functions that are missing the `fastmath` flag. "
+        msg += f"The function(s) are:\n {missing_fastmath}\n"
+        raise ValueError(msg)
+
+    return
+
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--perform", required=True)
+EXCEPTED_VALUES = ["check"]
+
+args = parser.parse_args()
+if args.perform not in EXCEPTED_VALUES:
+    raise ValueError("Invalid argument")
+
+if args.perform == "check":
+    check_fastmath()
+else:
+    pass

fastmath.py

@@ -93,10 +93,10 @@ check_print()
     fi
 }
 
-check_fastmath()
+fastmath()
 {
     echo "Checking Missing fastmath flags in njit functions"
-    ./check_fastmath.py
+    ./fastmath.py --perform check
     check_errs $?
 }
 
@@ -342,7 +342,7 @@ check_ray
 
 
 if [[ ! -z $NUMBA_DISABLE_JIT && ${NUMBA_DISABLE_JIT} -eq "0" ]]; then
-  check_fastmath
+  fastmath
 fi