Merge branch 'main' into fix-return-type

kaiko-ai · Aug 8, 2023 · 1deacdc · 1deacdc
2 parents 82064ce + 340fd3c
commit 1deacdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/typedspark/_core/dataset.py b/typedspark/_core/dataset.py
@@ -87,7 +87,7 @@ def _add_schema_metadata(self) -> None:
             self.schema[field.name].metadata = field.metadata
 
     @property
-    def typedspark_schema(self) -> T:
+    def typedspark_schema(self) -> Type[T]:
         """Returns the ``Schema`` of the ``DataSet``."""
         return self._schema_annotations  # type: ignore