From 23b76f416f91f645af4553fa515159c8081b68ee Mon Sep 17 00:00:00 2001
From: Sujeevan Rasaratnam <rsujeevan@gmail.com>
Date: Tue, 27 Aug 2024 08:50:52 -0400
Subject: [PATCH] add missing import (#210)

---
 models/mli/model_gam.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/models/mli/model_gam.py b/models/mli/model_gam.py
index 6df78178..9f188f43 100755
--- a/models/mli/model_gam.py
+++ b/models/mli/model_gam.py
@@ -191,6 +191,9 @@ def fit(self, X, y, sample_weight=None, eval_set=None, sample_weight_eval_set=No
                 raise IgnoreError('On entry to DLASCL parameter number') from e
             raise
         except:
+            import traceback
+            import sys
+
             t, v, tb = sys.exc_info()
             ex = ''.join(traceback.format_exception(t, v, tb))
             print(ex)