Feat - Fixed Circular Imports

configuration_autoencoder.py

@@ -5,7 +5,11 @@ Autoencoder configuration for Hugging Face Transformers.
 from dataclasses import dataclass
 from typing import Union
 
-from transformers import PretrainedConfig
+# Import PretrainedConfig in a way that avoids circular imports in some environments (e.g., Databricks)
+try:
+    from transformers.configuration_utils import PretrainedConfig
+except Exception:  # fallback
+    from transformers import PretrainedConfig
 from typing import List, Optional
 
 # Support both package-relative and flat import in HF remote code context

modeling_autoencoder.py

@@ -10,7 +10,13 @@ from dataclasses import dataclass
 import random
 import re
 
-from transformers import PreTrainedModel
+# Import PreTrainedModel in a way that avoids circular imports in some environments (e.g., Databricks)
+try:
+    from transformers.modeling_utils import PreTrainedModel
+except Exception:
+    # Fallback if direct path is unavailable
+    from transformers import PreTrainedModel
+
 from transformers.modeling_outputs import BaseModelOutput
 from transformers.utils import ModelOutput