From 4a4526f12f9ae53104f876fb4fd2ff783bfa5862 Mon Sep 17 00:00:00 2001
From: Lars Holmberg <lars.holmberg@svenskaspel.se>
Date: Tue, 12 Dec 2023 11:37:57 +0100
Subject: [PATCH] Type hint CSVReader class

---
 locust_plugins/csvreader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):