diff --git a/locust_plugins/csvreader.py b/locust_plugins/csvreader.py index 81a49a6..5064c5f 100644 --- a/locust_plugins/csvreader.py +++ b/locust_plugins/csvreader.py @@ -2,7 +2,7 @@ from typing import Iterator, Dict -class CSVReader: +class CSVReader(Iterator): "Read test data from csv file using an iterator" def __init__(self, file, **kwargs):