Reinstate CI data truncation

modal-labs · Feb 6, 2024 · 1bb3630 · 1bb3630
1 parent 0ae5b47
commit 1bb3630
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ci/prep_for_ci.py b/ci/prep_for_ci.py
@@ -18,10 +18,10 @@ def main(config: str, data: str):
     with open(config, "w") as f:
         yaml.dump(cfg, f)
 
-    # with open(data) as f:
-    #     data_truncated = f.readlines()[:2000]
-    # with open(data, "w") as f:
-    #     f.writelines(data_truncated)
+    with open(data) as f:
+        data_truncated = f.readlines()[:1000]
+    with open(data, "w") as f:
+        f.writelines(data_truncated)
 
 
 if __name__ == "__main__":