diff --git a/pyproject.toml b/pyproject.toml
index 50cf26e86..948e2d1b4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,8 +1,8 @@
 [project]
 name = "validmind"
+version = "2.9.4"
 description = "ValidMind Library"
 readme = "README.pypi.md"
-version = "2.9.3"
 requires-python = ">=3.9,<3.13"
 license = { file = "LICENSE" }
 authors = [
diff --git a/validmind/__version__.py b/validmind/__version__.py
index 4f1e6b1b5..dba9ee56f 100644
--- a/validmind/__version__.py
+++ b/validmind/__version__.py
@@ -1 +1 @@
-__version__ = "2.9.2"
+__version__ = "2.9.4"
diff --git a/validmind/client.py b/validmind/client.py
index 7f6d227c9..dcdb86386 100644
--- a/validmind/client.py
+++ b/validmind/client.py
@@ -11,7 +11,11 @@
 import numpy as np
 import pandas as pd
 import polars as pl
-import torch
+
+try:
+    import torch  # noqa: F401
+except Exception:
+    torch = None  # type: ignore  # noqa: F401
 
 from .api_client import log_input as log_input
 from .client_config import client_config