From 1ad1fbfe1e3a25df03cd89f30c466bbfc00c6539 Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Fri, 7 Apr 2023 11:01:30 +0000
Subject: [PATCH] dataset fix

---
 data/dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/data/dataset.py b/data/dataset.py
index 5643cab..ef8a87f 100644
--- a/data/dataset.py
+++ b/data/dataset.py
@@ -141,7 +141,7 @@ class Dataset(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 = { **item, **self.static_cache[frame.id] }
+                data = { **data, **self.static_cache[frame.id] }
                 return data
 
         separator = '_'
@@ -213,7 +213,7 @@ class Dataset(Dataset):
             self.cache[frame] = data
         
         data = { key: (value.to(self.device) if type(value) == torch.Tensor else value) for key, value in data.items()}
-        data = { **item, **self.static_cache[frame.id] }
+        data = { **data, **self.static_cache[frame.id] }
         return data