From 64b68bf9efa59ee826f5a05820ec43cc4a5cfda9 Mon Sep 17 00:00:00 2001
From: nanne-aben <47976799+nanne-aben@users.noreply.github.com>
Date: Wed, 4 Oct 2023 21:44:54 +0200
Subject: [PATCH] update

---
 typedspark/_core/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/typedspark/_core/dataset.py b/typedspark/_core/dataset.py
index 5fcf3d6..de4b14d 100644
--- a/typedspark/_core/dataset.py
+++ b/typedspark/_core/dataset.py
@@ -186,7 +186,7 @@ def __setattr__(self, name: str, value: Any) -> None:
         if name == "__orig_class__":
             orig_class_args = get_args(self.__orig_class__)
             if orig_class_args:
-                self._schema_annotations = orig_class_args[0]
+                self._schema_annotations: Type[_Schema] = orig_class_args[0]
                 validate_schema(
                     self._schema_annotations.get_structtype(),
                     deepcopy(self.schema),