fix version issue and spelling

mosaicml · May 24, 2024 · 313d52b · 313d52b
1 parent b1fbbef
commit 313d52b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/utils/test_exceptions.py b/tests/utils/test_exceptions.py
@@ -17,7 +17,10 @@ def create_exception_object(exception_name: str):
     )
     # get required arg types of exception class by inspecting its __init__ method
 
-    required_args = inspect.get_annotations(exception_class.__init__)
+    if hasattr(inspect, 'get_annotations'):
+        required_args = inspect.get_annotations(exception_class.__init__)
+    else:
+        required_args = exception_class.__init__.__annotations__  # python 3.9 and below
 
     # create a dictionary of required args with default values