From 2b62f419c292ca9c518daae55512fabc3fead4a4 Mon Sep 17 00:00:00 2001
From: Mehmet Deniz Birlikci <72691926+deniz-birlikci@users.noreply.github.com>
Date: Sun, 31 Dec 2023 04:22:45 +0300
Subject: [PATCH] Update setup.py (#269)

---
 setup.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/setup.py b/setup.py
index bdc9eb5..275b6fc 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,18 @@ import glob
 import os
 import subprocess
 
+import subprocess
+import sys
+
+def install_torch():
+    try:
+        import torch
+    except ImportError:
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "torch"])
+
+# Call the function to ensure torch is installed
+install_torch()
+
 import torch
 from setuptools import find_packages, setup
 from torch.utils.cpp_extension import CUDA_HOME, CppExtension, CUDAExtension