add missing imports

huggingface · Nov 5, 2022 · 1655b36 · 1655b36
1 parent 87740d5
commit 1655b36
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/setfit/data.py b/src/setfit/data.py
@@ -1,3 +1,5 @@
+import math
+import random
 from typing import TYPE_CHECKING, Dict, List, Tuple
 
 import pandas as pd