Fix error in documentation signature

FR-DC · Dec 28, 2023 · a78446a · a78446a
1 parent 60b835d
commit a78446a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/frdc/train/frdc_datamodule.py b/src/frdc/train/frdc_datamodule.py
@@ -36,7 +36,7 @@ class FRDCDataModule(LightningDataModule):
 
         Does not have the same performance as::
 
-            FRDCSSLDataModule(
+            FRDCDataModule(
                 train_lab_ds=train_lab_ds,
                 train_unl_ds=None,
                 ...