replace constant by function call

kaiko-ai · Nov 8, 2023 · 2bb40ea · 2bb40ea
1 parent e906099
commit 2bb40ea
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/typedspark/_schema/schema.py b/typedspark/_schema/schema.py
@@ -83,8 +83,9 @@ class A(Schema):
         """
         if (
             name.startswith("__")
-            or name in ["_attributes", "_abc_impl", "_is_protocol", "_is_runtime_protocol"]
+            or name == "_attributes"
             or name in cls._attributes
+            or name in dir(Protocol)
         ):
             return object.__getattribute__(cls, name)