From f9ac7fe3c38b92bdf1bc6ce70420e69586b66b6c Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Sat, 1 Jul 2023 10:44:32 +0200
Subject: [PATCH] fix static cache

---
 data/dataset.py | 4 ++--
 data/utils.py   | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/data/dataset.py b/data/dataset.py
index 98be66f..7ebc6d5 100644
--- a/data/dataset.py
+++ b/data/dataset.py
@@ -104,7 +104,7 @@ class StaticCache():
         return data
 
 
-    def __getitem__(self, frame) -> dict:
+    def __getitem__(self, frame: Frame) -> dict:
         if (frame.id, frame.garment) not in self.store:
             item = self.read(frame.id, frame.garment)
             stored_data = { key: torch.from_numpy(value).to('cpu') for key, value in item.items()}
@@ -164,7 +164,7 @@ class DatasetCloth3D(torch.utils.data.Dataset):
         if self.cache is not None:
             if frame in self.cache.keys():
                 data = { key: (value.to(self.device) if type(value) == torch.Tensor else value) for key, value in self.cache[frame].items()}
-                data = { **data, **self.static_cache[frame.id] }
+                data = { **data, **self.static_cache[frame] }
                 return data
 
         item = dict()
diff --git a/data/utils.py b/data/utils.py
index f6b796c..1e1596b 100644
--- a/data/utils.py
+++ b/data/utils.py
@@ -2,9 +2,9 @@ import tables as tb
 
 class Frame:
     def __init__(self, id: str, frame_nr: int, garment: str=''):
-        self.id = id
-        self.frame_nr = frame_nr
-        self.garment = garment
+        self.id: str = id
+        self.frame_nr: int = frame_nr
+        self.garment: str = garment
 
     def __hash__(self):
         return hash((self.id, self.frame_nr, self.garment))