tmp fixes for
@@ -92,10 +92,11 @@ class ImageWindowDatasetCSV(ImageWindowDataset):
returns a dictionary of sampler output tensor shapes
"""
assert self.window, 'Unknown output shapes: self.window not initialised'
- if self.csv_reader is not None:
- return {**self.window.tf_shapes, **self.csv_reader.tf_shapes}
- else:
- return {**self.window.tf_shapes}
+ # if self.csv_reader is not None:
+ # return {**self.window.tf_shapes, **self.csv_reader.tf_shapes}
+ # else:
+ # return {**self.window.tf_shapes}
+ return self.window.tf_shapes
@property
def tf_dtypes(self):
@@ -103,7 +104,8 @@ class ImageWindowDatasetCSV(ImageWindowDataset):
returns a dictionary of sampler output tensorflow dtypes
"""
assert self.window, 'Unknown output shapes: self.window not initialised'
- if self.csv_reader is not None:
- return {**self.window.tf_dtypes, **self.csv_reader.tf_dtypes}
- else:
- return {**self.window.tf_dtypes}
+ # if self.csv_reader is not None:
+ # return {**self.window.tf_dtypes, **self.csv_reader.tf_dtypes}
+ # else:
+ # return {**self.window.tf_dtypes}
+ return self.window.tf_dtypes
