Fix import error

ziatdinovmax · Oct 9, 2023 · 64b1d88 · 64b1d88
1 parent 3526b82
commit 64b1d88
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gpax/utils.py b/gpax/utils.py
@@ -8,6 +8,7 @@
 """
 
 import inspect
+import re
 from typing import Union, Dict, Type, List, Callable
 
 import jax