data.py change, date in paper.md

Author: homerjed

data/__init__.py

@@ -13,22 +13,23 @@
 def get_dataset(
     datasets_path: str,
     key: Key, 
-    config: ConfigDict
+    config: ConfigDict,
+    dataset_kwargs: dict
 ) -> ScalerDataset:
 
     dataset_name = config.dataset_name.lower()
 
     if dataset_name == "flowers":
-        dataset = flowers(key, n_pix=config.n_pix)
+        dataset = flowers(datasets_path, key, n_pix=config.n_pix, **dataset_kwargs)
     if dataset_name == "cifar10":
-        dataset = cifar10(datasets_path, key)
+        dataset = cifar10(datasets_path, key, **dataset_kwargs)
     if dataset_name == "mnist":
-        dataset = mnist(datasets_path, key)
+        dataset = mnist(datasets_path, key, **dataset_kwargs)
     if dataset_name == "moons":
         dataset = moons(key)
     if dataset_name == "grfs":
-        dataset = grfs(key, n_pix=config.n_pix)
+        dataset = grfs(key, n_pix=config.n_pix, **dataset_kwargs)
     if dataset_name == "quijote":
-        dataset = quijote(key, n_pix=config.n_pix, split=0.9)
+        dataset = quijote(key, n_pix=config.n_pix, split=0.9, **dataset_kwargs)
 
     return dataset

paper/paper.md

@@ -11,15 +11,15 @@ authors:
   - name: Jed Homer
     orcid: 0009-0002-0985-1437
     equal-contrib: true
-    affiliation: "1, 2" # (Multiple affiliations must be quoted)
+    affiliation: "1, 2" 
 affiliations:
  - name: University Observatory, Faculty for Physics, Ludwig-Maximilians-Universität München, Scheinerstrasse 1, München, Deustchland.
    index: 1
    ror: 00hx57361
  - name: Munich Center for Machine Learning.
    index: 2
    ror: 00hx57361
-date: 1 October 2024
+date: 25 January 2025
 bibliography: paper.bib
 
 # Optional fields if submitting to a AAS journal too, see this blog post: