File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1600,7 +1600,7 @@ def _cachecheck(self, item_transformed):
16001600 if hashfile is not None and hashfile .is_file (): # cache hit
16011601 with cp .cuda .Device (self .device ):
16021602 if isinstance (item_transformed , dict ):
1603- item : dict [Any , Any ] = {} # type:ignore
1603+ item : dict [Any , Any ] = {}
16041604 for k in item_transformed :
16051605 meta_k = self ._load_meta_cache (meta_hash_file_name = f"{ hashfile .name } -{ k } -meta" )
16061606 item [k ] = kvikio_numpy .fromfile (f"{ hashfile } -{ k } " , dtype = meta_k ["dtype" ], like = cp .empty (()))
@@ -1688,4 +1688,4 @@ def _load_meta_cache(self, meta_hash_file_name):
16881688 if meta_hash_file_name in self ._meta_cache :
16891689 return self ._meta_cache [meta_hash_file_name ]
16901690 else :
1691- return torch .load (self .cache_dir / meta_hash_file_name ) # type:ignore
1691+ return torch .load (self .cache_dir / meta_hash_file_name )
You can’t perform that action at this time.
0 commit comments