From f46cfe933fd1c794c9290b79caa74a641b465b6f Mon Sep 17 00:00:00 2001
From: Benjamin Paul Jaeger <benjamin.jaeger@cern.ch>
Date: Wed, 25 May 2022 11:20:37 -0700
Subject: [PATCH] Adapt tests 3
---
freeforestml/tests/test_model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/freeforestml/tests/test_model.py b/freeforestml/tests/test_model.py
index d019f42..c596b98 100644
--- a/freeforestml/tests/test_model.py
+++ b/freeforestml/tests/test_model.py
@@ -908,7 +908,7 @@ class HepNetTestCase(unittest.TestCase):
df["is_sig"] = (df.fpid == 1)
df["is_ztt"] = (df.fpid == 0)
- net.fit(df.compute(), epochs=5, verbose=0)
+ net.fit(df, epochs=5, verbose=0)
fd, path = tempfile.mkstemp()
--
GitLab