diff --git a/tabpfn_client/client.py b/tabpfn_client/client.py
index 746a3ed..4fe1c4c 100644
--- a/tabpfn_client/client.py
+++ b/tabpfn_client/client.py
@@ -6,7 +6,6 @@
 import logging
 from importlib.metadata import version, PackageNotFoundError
 import numpy as np
-from enum import Enum
 from omegaconf import OmegaConf
 import json
 from typing import Literal