From 4e13e2a7d579845faf3ae6f01b3e94c858f4f4f9 Mon Sep 17 00:00:00 2001
From: zahid-syed <zahid.s2618@gmail.com>
Date: Mon, 19 Feb 2024 20:58:22 -0500
Subject: [PATCH] tests and linting fix
---
tests/unit/test_layer.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 00bad4ff..3f2c413f 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -432,13 +432,13 @@ class TestLayerFit:
# unpack test data
X, y = zip(*test_data)
# evaluate
- route_layer.evaluate(X=X, y=y)
+ route_layer.evaluate(X=X, y=y, batch_size=int(len(test_data) / 5))
def test_fit(self, openai_encoder, routes, test_data):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
# unpack test data
X, y = zip(*test_data)
- route_layer.fit(X=X, y=y)
+ route_layer.fit(X=X, y=y, batch_size=int(len(test_data) / 5))
# Add more tests for edge cases and error handling as needed.
--
GitLab