From e7c1882b0372b9bf008a4c36c11853091d1a96c8 Mon Sep 17 00:00:00 2001
From: Julia Putko <juliaputko@gmail.com>
Date: Tue, 20 Aug 2024 15:55:53 -0500
Subject: [PATCH] added torch to ml_defaults

---
 smartsim/_core/_install/buildenv.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/smartsim/_core/_install/buildenv.py b/smartsim/_core/_install/buildenv.py
index 844d86f4d..c147ee446 100644
--- a/smartsim/_core/_install/buildenv.py
+++ b/smartsim/_core/_install/buildenv.py
@@ -211,6 +211,7 @@ def ml_extras_required(self) -> t.Dict[str, t.List[str]]:
         """Optional ML/DL dependencies we suggest for the user.
         """
         ml_defaults = {
+            "torch": self.TORCH,
             "tensorflow": self.TENSORFLOW,
             "onnx": self.ONNX,
             "skl2onnx": "1.16.0",