From 1ff9bbaa613de00f36a0625f28529df95a6510fb Mon Sep 17 00:00:00 2001
From: Roelof van Dijk <3604013+roelofvandijk@users.noreply.github.com>
Date: Thu, 27 Jun 2024 20:29:47 +0200
Subject: [PATCH] ruff: close file handle (#5180)

* close file handle

* some more open file handles

* must stay open

* remove this close, stays open
---
 tinygrad/engine/schedule.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py
index 7dd555bcb7..469c7d771c 100644
--- a/tinygrad/engine/schedule.py
+++ b/tinygrad/engine/schedule.py
@@ -316,7 +316,7 @@ def create_schedule_with_vars(outs:List[LazyBuffer], seen:Optional[Set[LazyBuffe
   if SAVE_SCHEDULE:
     def _save():
       print(f"saving {len(SCHEDULES)} schedule graphs to", fp:=getenv("SAVE_SCHEDULE_PATH", "schedule.pkl"))
-      pickle.dump(SCHEDULES, open(fp, "wb"))
+      with open(fp, "wb") as f: pickle.dump(SCHEDULES, f)
     if len(SCHEDULES) == 0: atexit.register(_save)
     SCHEDULES.extend((ps.ast for ps in prescheduled.values()) if getenv("CAPTURE_AST") else [(graph, prescheduled)])
   # confirm everything was scheduled correctly