(self, ar_frames, resolutions, metadata_dataset, directory_config, cache_base, round_to_multiple, directory_dataset)
| 397 | |
| 398 | class ARBucketDataset: |
| 399 | def __init__(self, ar_frames, resolutions, metadata_dataset, directory_config, cache_base, round_to_multiple, directory_dataset): |
| 400 | self.ar_frames = ar_frames |
| 401 | self.resolutions = resolutions |
| 402 | self.metadata_dataset = metadata_dataset |
| 403 | self.directory_config = directory_config |
| 404 | self.size_buckets = [] |
| 405 | self.path = Path(directory_config['path']) |
| 406 | self.cache_base = cache_base |
| 407 | self.cache_dir = cache_base / f'ar_frames_{bucket_suffix(self.ar_frames)}' |
| 408 | self.round_to_multiple = round_to_multiple |
| 409 | self.directory_dataset = directory_dataset |
| 410 | os.makedirs(self.cache_dir, exist_ok=True) |
| 411 | |
| 412 | def get_size_bucket_datasets(self): |
| 413 | return self.size_buckets |
nothing calls this directly
no test coverage detected