From 720ea65d44d97d14efd921c86a92ed42b0df655e Mon Sep 17 00:00:00 2001
From: David Hall <dlwh@stanford.edu>
Date: Thu, 8 Feb 2024 09:56:58 -0800
Subject: [PATCH] missed an import

---
 src/levanter/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/levanter/trainer.py b/src/levanter/trainer.py
index a98e0c10b..8cc4b4410 100644
--- a/src/levanter/trainer.py
+++ b/src/levanter/trainer.py
@@ -5,6 +5,7 @@
 import os
 import sys
 import typing
+import warnings
 from dataclasses import dataclass
 from functools import cached_property
 from pathlib import Path