Update types.py

types.py

@@ -1,3 +1,5 @@
+from custom_types import SomeClass  # Updated import
+
 class Code:
     def __init__(self, code: str, language: str):
         self.code = code