chore: import upstream snapshot with attribution

python/ray/tests/kuberay/scripts/non_terminated_nodes_count.py

@@ -0,0 +1,26 @@
+import ray
+from ray.autoscaler._private.kuberay.autoscaling_config import _generate_provider_config
+from ray.autoscaler._private.providers import _get_node_provider
+
+
+@ray.remote
+def count_non_terminated_nodes() -> int:
+    """Get the count of non terminated nodes for the Ray cluster raycluster-autoscaler
+    in namespace default.
+    """
+    provider_config = _generate_provider_config(ray_cluster_namespace="default")
+    kuberay_node_provider = _get_node_provider(
+        provider_config=provider_config, cluster_name="raycluster-autoscaler"
+    )
+    nodes = kuberay_node_provider.non_terminated_nodes({})
+    return len(nodes)
+
+
+def main() -> int:
+    return ray.get(count_non_terminated_nodes.remote())
+
+
+if __name__ == "__main__":
+    ray.init("auto")
+    out = main()
+    print(out)