Double the size of the example dataset

golmschenk · Feb 1, 2024 · 63a21ed · 63a21ed
1 parent 4cb532a
commit 63a21ed
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/examples/download_spoc_transit_light_curves.py b/examples/download_spoc_transit_light_curves.py
@@ -31,32 +31,32 @@ def main():
         tic_ids=positive_train_tic_ids,
         download_directory=Path('data/spoc_transit_experiment/train/positives'),
         sectors=sectors,
-        limit=1000)
+        limit=2000)
     download_spoc_light_curves_for_tic_ids(
         tic_ids=negative_train_tic_ids,
         download_directory=Path('data/spoc_transit_experiment/train/negatives'),
         sectors=sectors,
-        limit=3000)
+        limit=6000)
     download_spoc_light_curves_for_tic_ids(
         tic_ids=positive_validation_tic_ids,
         download_directory=Path('data/spoc_transit_experiment/validation/positives'),
         sectors=sectors,
-        limit=100)
+        limit=200)
     download_spoc_light_curves_for_tic_ids(
         tic_ids=negative_validation_tic_ids,
         download_directory=Path('data/spoc_transit_experiment/validation/negatives'),
         sectors=sectors,
-        limit=300)
+        limit=600)
     download_spoc_light_curves_for_tic_ids(
         tic_ids=positive_test_tic_ids,
         download_directory=Path('data/spoc_transit_experiment/test/positives'),
         sectors=sectors,
-        limit=100)
+        limit=200)
     download_spoc_light_curves_for_tic_ids(
         tic_ids=negative_test_tic_ids,
         download_directory=Path('data/spoc_transit_experiment/test/negatives'),
         sectors=sectors,
-        limit=300)
+        limit=600)
 
 
 if __name__ == '__main__':

diff --git a/src/qusi/infer_session.py b/src/qusi/infer_session.py
@@ -22,7 +22,7 @@ def infer_session(infer_datasets: List[FiniteStandardLightCurveDataset], model:
     return results
 
 
-def get_device():
+def get_device() -> Device:
     if torch.cuda.is_available():
         device = torch.device('cuda')
     else: