From 15854641995680d36056246acec0bc5456ad0b54 Mon Sep 17 00:00:00 2001
From: Oscar Gustafsson <oscar.gustafsson@gmail.com>
Date: Fri, 5 May 2023 12:35:17 +0200
Subject: [PATCH] Remove lambda to enable saving

---
 b_asic/schedule.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/b_asic/schedule.py b/b_asic/schedule.py
index 925615cb..237430a3 100644
--- a/b_asic/schedule.py
+++ b/b_asic/schedule.py
@@ -966,7 +966,7 @@ class Schedule:
 
     def _reset_y_locations(self) -> None:
         """Reset all the y-locations in the schedule to None"""
-        self._y_locations = defaultdict(lambda: None)
+        self._y_locations = defaultdict(_y_locations_default)
 
     def plot(self, ax: Axes, operation_gap: Optional[float] = None) -> None:
         """
-- 
GitLab