add self and tomllib imports

ProKil · Dec 2, 2024 · c2d3dfc · c2d3dfc
1 parent b9eff3b
commit c2d3dfc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/aact/utils/tomllib.py b/src/aact/utils/tomllib.py
@@ -2,6 +2,8 @@
 
 
 if sys.version_info >= (3, 11):
-    pass
+    import tomllib
 else:
-    pass
+    import tomlkit as tomllib
+
+__all__ = ["tomllib"]
diff --git a/src/aact/utils/types.py b/src/aact/utils/types.py
@@ -2,6 +2,9 @@
 
 
 if sys.version_info >= (3, 11):
-    pass
+    from typing import Self
 else:
-    pass
+    from typing_extensions import Self
+
+
+__all__ = ["Self"]