Skip to content
Snippets Groups Projects
Commit 4e13e2a7 authored by zahid-syed's avatar zahid-syed
Browse files

tests and linting fix

parent ea09babc
No related branches found
No related tags found
No related merge requests found
...@@ -432,13 +432,13 @@ class TestLayerFit: ...@@ -432,13 +432,13 @@ class TestLayerFit:
# unpack test data # unpack test data
X, y = zip(*test_data) X, y = zip(*test_data)
# evaluate # 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): def test_fit(self, openai_encoder, routes, test_data):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes) route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
# unpack test data # unpack test data
X, y = zip(*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. # Add more tests for edge cases and error handling as needed.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment