From 627881be2e9d5a88837809604db69b838425d54d Mon Sep 17 00:00:00 2001
From: Kevin Wang <kevin@eventualcomputing.com>
Date: Fri, 22 Nov 2024 13:11:18 -0800
Subject: [PATCH] fixes

---
 daft/daft/__init__.pyi             | 7 -------
 src/common/io-config/src/python.rs | 2 +-
 2 files changed, 1 insertion(+), 8 deletions(-)

diff --git a/daft/daft/__init__.pyi b/daft/daft/__init__.pyi
index b821f58115..6da06bb7f6 100644
--- a/daft/daft/__init__.pyi
+++ b/daft/daft/__init__.pyi
@@ -638,13 +638,6 @@ class IOConfig:
         gcs: GCSConfig | None = None,
         http: HTTPConfig | None = None,
     ): ...
-    @staticmethod
-    def from_json(input: str) -> IOConfig:
-        """
-        Recreate an IOConfig from a JSON string.
-        """
-        ...
-
     def replace(
         self,
         s3: S3Config | None = None,
diff --git a/src/common/io-config/src/python.rs b/src/common/io-config/src/python.rs
index 41784ae062..e161e052e6 100644
--- a/src/common/io-config/src/python.rs
+++ b/src/common/io-config/src/python.rs
@@ -133,7 +133,7 @@ pub struct GCSConfig {
 ///     >>> io_config = IOConfig(s3=S3Config(key_id="xxx", access_key="xxx", num_tries=10), azure=AzureConfig(anonymous=True), gcs=GCSConfig(...))
 ///     >>> daft.read_parquet(["s3://some-path", "az://some-other-path", "gs://path3"], io_config=io_config)
 #[derive(Clone, Default, Serialize, Deserialize)]
-#[pyclass]
+#[pyclass(module = "daft.daft")]
 pub struct IOConfig {
     pub config: config::IOConfig,
 }