add snowflake.cortex to setup

Signed-off-by: Labanya Mukhopadhyay <[email protected]>
snowflakedb · Jan 28, 2025 · c8b3d10 · c8b3d10
1 parent 90acd1e
commit c8b3d10
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -202,6 +202,7 @@ def run(self):
             "scikit-learn",  # Snowpark pandas 3rd party library testing
             # plotly version restricted due to foreseen change in query counts in version 6.0.0+
             "plotly<6.0.0",  # Snowpark pandas 3rd party library testing
+            "snowflake.cortex",
         ],
         "localtest": [
             "pandas",