Merge pull request #709 from Andrew-Robertson/master

fix(perf): only load the necessary parts of each dataset
galacticusorg · Sep 25, 2024 · e5bbec1 · e5bbec1
2 parents 062da28 + 0d78c7d
commit e5bbec1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/aux/extractSingleTree.py b/scripts/aux/extractSingleTree.py
@@ -62,8 +62,8 @@ def restricted_integer(x):
     if name == "particleIndexStart" or name == "particleIndexCount":
         continue
     print("   "+name)
-    dataset    = fromFile[halosName][name][:]
-    toFile["forestHalos"].create_dataset(name,data=dataset[start[0]:end[0],...])
+    dataset    = fromFile[halosName][name][start[0]:end[0],...]
+    toFile["forestHalos"].create_dataset(name,data=dataset)
 print("...done")
 
 # Create the forestIndex group.