We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f4d5f44 commit ec22aa8Copy full SHA for ec22aa8
test/utils.py
@@ -392,7 +392,10 @@ def get_frame_data_by_index(
392
if stream_index is None:
393
stream_index = self.default_stream_index
394
395
- tensor_file_path = f"{self.get_base_path_by_index(idx, stream_index=stream_index, filters=filters)}.pt"
+ base_path = self.get_base_path_by_index(
396
+ idx, stream_index=stream_index, filters=filters
397
+ )
398
+ tensor_file_path = f"{base_path}.pt"
399
return torch.load(tensor_file_path, weights_only=True).permute(2, 0, 1)
400
401
def get_frame_data_by_range(
0 commit comments