dataset_class param in GeoDataModule

[roybenhayun]

see in answer for another question:

I would recommend extending GeoDataModule or NonGeoDataModule. That's what they're there for.

what should be the dataset_class param to GeoDataModule?
for example, in the training-example, the module extends LightningDataModule and creates the dataset in the setup method. there is no class.
how would that be refactored?

[roybenhayun]

just to note that the training-example is from few releases ago. wondering how this change will apply to either 0.4.1 and 0.5.0

[adamjstewart]

From the docs:

dataset_class – Class used to instantiate a new dataset.

You can also see examples of this in https://github.com/microsoft/torchgeo/tree/main/torchgeo/datamodules

[roybenhayun]

tried to go over few of the builtin datamodules also in the context of building an ObjectDetectionTask.
with ObjectDetectionTask specifically, there seems to be a conflict in some assertion.
some working code samples I have don't use datamodules.

without some explaining post or a tutorial coupled with working examples it is not straightforward to understand from the builtin classes alone.