From 69de4dacaf72055de7830862a5a583e9031e7e45 Mon Sep 17 00:00:00 2001
From: FNTwin <cris.gabellini@gmail.com>
Date: Tue, 16 Jul 2024 12:12:29 -0400
Subject: [PATCH] tuple->Tuple

---
 openqdc/datasets/structure.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/openqdc/datasets/structure.py b/openqdc/datasets/structure.py
index 8adfce4..f6dc077 100644
--- a/openqdc/datasets/structure.py
+++ b/openqdc/datasets/structure.py
@@ -2,7 +2,7 @@
 from abc import ABC, abstractmethod
 from os import PathLike
 from os.path import join as p_join
-from typing import Callable, Dict, List, Optional, Union
+from typing import Callable, Dict, List, Optional, Tuple, Union
 
 import numpy as np
 import zarr
@@ -108,7 +108,7 @@ def load_data(
         preprocess_path: Union[str, PathLike],
         data_keys: List[str],
         data_types: Dict[str, np.dtype],
-        data_shapes: Dict[str, tuple[int, int]],
+        data_shapes: Dict[str, Tuple[int, int]],
         extra_data_keys: List[str],
         overwrite: bool,
     ):