Commit
•
32e4e72
1
Parent(s):
3538a75
deleting the pipelines in the end of the test
Browse files
tests.py
CHANGED
|
@@ -61,6 +61,7 @@ def test_create_pipeline(model_name: str):
|
|
| 61 |
assert pipeline.model_name == model_name
|
| 62 |
assert pipeline.wrapped_model.group_size == 5
|
| 63 |
assert pipeline.wrapped_model.end_of_sentence_stop is False
|
|
|
|
| 64 |
|
| 65 |
|
| 66 |
if __name__ == "__main__":
|
|
|
|
| 61 |
assert pipeline.model_name == model_name
|
| 62 |
assert pipeline.wrapped_model.group_size == 5
|
| 63 |
assert pipeline.wrapped_model.end_of_sentence_stop is False
|
| 64 |
+
del pipeline
|
| 65 |
|
| 66 |
|
| 67 |
if __name__ == "__main__":
|