Don't check if identity is dict

basedformer/dataset.py

@@ -81,8 +81,6 @@ class ShardedImageDataset(data.Dataset):
         self.pointer_lookup = {}
         for t in self.index:
             offset, length, id = t
-            if type(id) == dict: # if id is identity
-                id = id['id']
             self.pointer_lookup[id] = (offset, length)
 
         #make so metadata is shardable by world_size(num_gpus)