Fix EnvBaseMixins issues when subclassing with other generics

AllenInstitute · Oct 30, 2024 · 2e2e847 · 2e2e847
1 parent 94701eb
commit 2e2e847
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/aibs_informatics_core/env.py b/src/aibs_informatics_core/env.py
@@ -223,7 +223,9 @@ def _env_base_class(cls) -> Type[E]:
 
 
 class EnvBaseMixins(EnvBaseMixinsBase[EnvBase]):
-    pass
+    @classmethod
+    def _env_base_class(cls) -> Type[EnvBase]:
+        return EnvBase
 
 
 class EnvBaseEnumMixins: