moved the not implemented versions of Checked* classes in the beginning of checked.py otherwise PyCharm do not provide auto-completion for methods imported from Session

Author: alixdamman

larray/core/checked.py

@@ -22,7 +22,20 @@ class NotLoaded:
 except ImportError:
     pydantic = None
 
-if pydantic:
+if not pydantic:
+    def CheckedArray(axes: AxisCollection, dtype: np.dtype = float) -> Type[Array]:
+        raise NotImplementedError("CheckedArray cannot be used because pydantic is not installed")
+
+    class CheckedSession:
+        def __init__(self, *args, **kwargs):
+            raise NotImplementedError("CheckedSession class cannot be instantiated "
+                                      "because pydantic is not installed")
+
+    class CheckedParameters:
+        def __init__(self, *args, **kwargs):
+            raise NotImplementedError("CheckedParameters class cannot be instantiated "
+                                      "because pydantic is not installed")
+else:
     from pydantic.fields import ModelField
     from pydantic.class_validators import Validator
     from pydantic.main import BaseConfig
@@ -518,17 +531,3 @@ class Config:
             # whether or not models are faux-immutable, i.e. whether __setattr__ is allowed.
             # (default: True)
             allow_mutation = False
-
-else:
-    def CheckedArray(axes: AxisCollection, dtype: np.dtype = float) -> Type[Array]:
-        raise NotImplementedError("CheckedArray cannot be used because pydantic is not installed")
-
-    class CheckedSession:
-        def __init__(self, *args, **kwargs):
-            raise NotImplementedError("CheckedSession class cannot be instantiated "
-                                      "because pydantic is not installed")
-
-    class CheckedParameters:
-        def __init__(self, *args, **kwargs):
-            raise NotImplementedError("CheckedParameters class cannot be instantiated "
-                                      "because pydantic is not installed")