Move type alias inside TYPE_CHECKING check

Author: pchlenski

manify/utils/benchmarks.py

@@ -12,6 +12,32 @@
 if TYPE_CHECKING:
     from beartype.typing import Literal, TypeAlias
     from jaxtyping import Float, Real
+
+    MODELTYPE: TypeAlias = Literal[
+        "sklearn_dt",
+        "sklearn_rf",
+        "product_dt",
+        "product_rf",
+        "tangent_dt",
+        "tangent_rf",
+        "knn",
+        "ps_perceptron",
+        "svm",
+        "ps_svm",
+        "kappa_mlp",
+        "tangent_mlp",
+        "ambient_mlp",
+        "tangent_gcn",
+        "ambient_gcn",
+        "kappa_gcn",
+        "ambient_mlr",
+        "tangent_mlr",
+        "kappa_mlr",
+        "single_manifold_rf",
+    ]
+    SCORETYPE: TypeAlias = Literal["accuracy", "f1-micro", "f1-macro", "mse", "percent_rmse", "time"]
+    TASKTYPE: TypeAlias = Literal["classification", "regression", "link_prediction"]
+
 from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
 from sklearn.linear_model import SGDClassifier, SGDRegressor
 from sklearn.metrics import accuracy_score, f1_score, mean_squared_error, root_mean_squared_error
@@ -26,31 +52,6 @@
 from ..predictors.perceptron import ProductSpacePerceptron
 from ..predictors.svm import ProductSpaceSVM
 
-MODELTYPE: TypeAlias = Literal[
-    "sklearn_dt",
-    "sklearn_rf",
-    "product_dt",
-    "product_rf",
-    "tangent_dt",
-    "tangent_rf",
-    "knn",
-    "ps_perceptron",
-    "svm",
-    "ps_svm",
-    "kappa_mlp",
-    "tangent_mlp",
-    "ambient_mlp",
-    "tangent_gcn",
-    "ambient_gcn",
-    "kappa_gcn",
-    "ambient_mlr",
-    "tangent_mlr",
-    "kappa_mlr",
-    "single_manifold_rf",
-]
-SCORETYPE: TypeAlias = Literal["accuracy", "f1-micro", "f1-macro", "mse", "percent_rmse", "time"]
-TASKTYPE: TypeAlias = Literal["classification", "regression", "link_prediction"]
-
 
 def _score(
     _X: Float[torch.Tensor | np.ndarray, "batch dim"] | None,