Fix: issue #1297: datasaver does not work with __future__.annotations

hamilton/function_modifiers/adapters.py

@@ -878,7 +878,7 @@ def validate(self, fn: Callable):
                 f"Function: {fn.__qualname__} must have a return annotation."
             )
         origin = typing.get_origin(return_annotation)
-        if return_annotation is not dict and origin is not dict:
+        if return_annotation not in (dict, Dict, "dict", "dict_") and origin not in (dict, Dict, "dict", "dict_"):
             raise InvalidDecoratorException(f"Function: {fn.__qualname__} must return a dict.")
 
     def generate_nodes(self, fn: Callable, config) -> list[node.Node]:

tests/function_modifiers/test_adapters.py

@@ -14,6 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+from __future__ import annotations
 
 import dataclasses
 from collections import Counter