From b4aaa8aaaaa8898a87165b6277b81f778a666e19 Mon Sep 17 00:00:00 2001
From: Deependra Patel <pateldeependra06@gmail.com>
Date: Thu, 1 Jan 2026 23:17:00 +0700
Subject: [PATCH] feat: Define "extra" optional dependencies with ml, starting
 with installing dataproc-ml

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

diff --git a/setup.py b/setup.py
index b79e3ab..74a75fd 100644
--- a/setup.py
+++ b/setup.py
@@ -35,4 +35,7 @@
         "tqdm>=4.67",
         "websockets>=14.0",
     ],
+    extras_require={
+        "ml": ["dataproc-ml>=1.0.0-rc1,<=1.0"],
+    },
 )