make test fix

Author: ritaagarwala-sf

src/datacustomcode/config.py

@@ -36,7 +36,10 @@
 # This lets all readers and writers to be findable via config
 from datacustomcode.io import *  # noqa: F403
 from datacustomcode.io.base import BaseDataAccessLayer
+from datacustomcode.io.reader.base import BaseDataCloudReader  # noqa: TCH002
+from datacustomcode.io.writer.base import BaseDataCloudWriter  # noqa: TCH002
 from datacustomcode.proxy.base import BaseProxyAccessLayer
+from datacustomcode.proxy.client.base import BaseProxyClient  # noqa: TCH002
 from datacustomcode.spark.base import BaseSparkSessionProvider
 
 DEFAULT_CONFIG_NAME = "config.yaml"
@@ -45,10 +48,6 @@
 if TYPE_CHECKING:
     from pyspark.sql import SparkSession
 
-    from datacustomcode.io.reader.base import BaseDataCloudReader
-    from datacustomcode.io.writer.base import BaseDataCloudWriter
-    from datacustomcode.proxy.client.base import BaseProxyClient
-
 
 class ForceableConfig(BaseModel):
     force: bool = Field(