Remove irrelevant IDEs

Author: medb

google/cloud/dataproc_spark_connect/environment.py

@@ -95,11 +95,6 @@ def is_hex() -> bool:
     return os.getenv("HEX_PROJECT_ID") is not None
 
 
-def is_jetski() -> bool:
-    """True if running in JetSki."""
-    return os.getenv("JETSKI_VERSION") is not None
-
-
 def is_polynote() -> bool:
     """True if running in Polynote."""
     return os.getenv("POLYNOTE_VERSION") is not None
@@ -152,14 +147,13 @@ def get_client_environment_label() -> str:
       9. GitHub Codespaces ("codespaces")
       10. Google Cloud Shell ("cloud-shell")
       11. Hex ("hex")
-      12. JetSki ("jetski")
-      13. Polynote ("polynote")
-      14. VS Code ("vscode")
-      15. JetBrains IDE ("jetbrains")
-      16. Spyder ("spyder")
-      17. Eclipse ("eclipse")
-      18. Jupyter ("jupyter")
-      19. Unknown ("unknown")
+      12. Polynote ("polynote")
+      13. VS Code ("vscode")
+      14. JetBrains IDE ("jetbrains")
+      15. Spyder ("spyder")
+      16. Eclipse ("eclipse")
+      17. Jupyter ("jupyter")
+      18. Unknown ("unknown")
     """
     checks: List[Tuple[Callable[[], bool], str]] = [
         (is_colab_enterprise, "colab-enterprise"),
@@ -173,7 +167,6 @@ def get_client_environment_label() -> str:
         (is_codespaces, "codespaces"),
         (is_cloud_shell, "cloud-shell"),
         (is_hex, "hex"),
-        (is_jetski, "jetski"),
         (is_polynote, "polynote"),
         (is_vscode, "vscode"),
         (is_jetbrains_ide, "jetbrains"),

tests/unit/test_environment.py

@@ -144,14 +144,6 @@ def test_is_hex_false(self):
         os.environ.pop("HEX_PROJECT_ID", None)
         self.assertFalse(environment.is_hex())
 
-    def test_is_jetski_true(self):
-        os.environ["JETSKI_VERSION"] = "1.0"
-        self.assertTrue(environment.is_jetski())
-
-    def test_is_jetski_false(self):
-        os.environ.pop("JETSKI_VERSION", None)
-        self.assertFalse(environment.is_jetski())
-
     def test_is_polynote_true(self):
         os.environ["POLYNOTE_VERSION"] = "1.0"
         self.assertTrue(environment.is_polynote())
@@ -170,6 +162,14 @@ def test_is_eclipse_false(self):
                 os.environ.pop(k)
         self.assertFalse(environment.is_eclipse())
 
+    def test_is_antigravity_true(self):
+        os.environ["ANTIGRAVITY_IDE"] = "1.0"
+        self.assertTrue(environment.is_antigravity())
+
+    def test_is_antigravity_false(self):
+        os.environ.pop("ANTIGRAVITY_IDE", None)
+        self.assertFalse(environment.is_antigravity())
+
     def test_is_jetbrains_ide_true(self):
         os.environ["TERMINAL_EMULATOR"] = "JetBrains term"
         self.assertTrue(environment.is_jetbrains_ide())
@@ -235,10 +235,6 @@ def test_is_jetbrains_ide_false_env_var_not_jetbrains(self):
         "google.cloud.dataproc_spark_connect.environment.is_hex",
         return_value=False,
     )
-    @mock.patch(
-        "google.cloud.dataproc_spark_connect.environment.is_jetski",
-        return_value=False,
-    )
     @mock.patch(
         "google.cloud.dataproc_spark_connect.environment.is_polynote",
         return_value=False,
@@ -259,6 +255,10 @@ def test_is_jetbrains_ide_false_env_var_not_jetbrains(self):
         "google.cloud.dataproc_spark_connect.environment.is_eclipse",
         return_value=False,
     )
+    @mock.patch(
+        "google.cloud.dataproc_spark_connect.environment.is_antigravity",
+        return_value=False,
+    )
     @mock.patch(
         "google.cloud.dataproc_spark_connect.environment.is_jupyter",
         return_value=False,
@@ -379,10 +379,6 @@ def test_get_client_environment_label_kaggle(self, *mocks):
         "google.cloud.dataproc_spark_connect.environment.is_hex",
         return_value=False,
     )
-    @mock.patch(
-        "google.cloud.dataproc_spark_connect.environment.is_jetski",
-        return_value=False,
-    )
     @mock.patch(
         "google.cloud.dataproc_spark_connect.environment.is_polynote",
         return_value=False,
@@ -441,10 +437,6 @@ def test_get_client_environment_label_vscode(self, *mocks):
         "google.cloud.dataproc_spark_connect.environment.is_hex",
         return_value=False,
     )
-    @mock.patch(
-        "google.cloud.dataproc_spark_connect.environment.is_jetski",
-        return_value=False,
-    )
     @mock.patch(
         "google.cloud.dataproc_spark_connect.environment.is_polynote",
         return_value=False,