(self, num_workers=0, batch_size=1000,data_root='/home/kklemmer/sphericalharmonics/data/era5', label_key='t2m')
| 79 | |
| 80 | class ERA5DataModule(pl.LightningDataModule): |
| 81 | def __init__(self, num_workers=0, batch_size=1000,data_root='/home/kklemmer/sphericalharmonics/data/era5', label_key='t2m'): |
| 82 | super().__init__() |
| 83 | self.batch_size=batch_size |
| 84 | self.data_root=data_root |
| 85 | self.num_workers=num_workers |
| 86 | self.label_key = label_key |
| 87 | |
| 88 | |
| 89 | def setup(self, stage: str): |
nothing calls this directly
no outgoing calls
no test coverage detected