Add comment for issue #1

camlsys · Mar 18, 2021 · 257d6bc · 257d6bc
1 parent 5f05601
commit 257d6bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/reddit_binary/dataset.py b/reddit_binary/dataset.py
@@ -67,7 +67,8 @@ def get_dataset(path, name, sparse=True, cleaned=False, DQ=None):
         dq_transform = ProbabilisticHighDegreeMask(
             DQ["prob_mask_low"], min(DQ["prob_mask_low"] + DQ["prob_mask_change"], 1.0)
         )
-
+        # NOTE: see issue #1 if you are customizing for your own dataset
+        # dataset.transform may be None (not the case here)
         dataset.transform = T.Compose([dataset.transform, dq_transform])
 
     return dataset