pytorch · clee2000 · Jan 3, 2025 · Jan 3, 2025 · Jan 3, 2025 · Jan 3, 2025
diff --git a/tools/torchci/requirements.txt b/tools/torchci/requirements.txt
@@ -1,6 +1,5 @@
 setuptools==70.0.0
 requests
-rockset==1.0.3
 boto3==1.19.12
 pytest==7.2.0
 clickhouse-connect==0.7.16
diff --git a/tools/torchci/rockset_utils.py b/tools/torchci/rockset_utils.py
diff --git a/torchci/.env.example b/torchci/.env.example
@@ -2,10 +2,6 @@
 # these keys can be found on Vercel.  Please see the README about getting access
 # to Vercel.
 
-# Rockset API key for talking to our data backend. You can generate one from the
-# Rockset console (https://console.rockset.com/apikeys).
-ROCKSET_API_KEY=
-
 # GitHub App credentials for probot. Only needed if testing bot stuff locally.
 PRIVATE_KEY=
 APP_ID=

diff --git a/torchci/package.json b/torchci/package.json
@@ -32,7 +32,6 @@
     "@octokit/webhooks": "^13.3.0",
     "@octokit/webhooks-types": "^7.6.1",
     "@opensearch-project/opensearch": "^2.3.1",
-    "@rockset/client": "^0.8.9",
     "@types/minimist": "^1.2.2",
     "@vercel/analytics": "^1.3.2",
     "ansicolor": "^1.1.100",

diff --git a/torchci/rockset.config.json b/torchci/rockset.config.json
diff --git a/torchci/rockset/benchmarks/__sql/oss_ci_benchmark_branches.sql b/torchci/rockset/benchmarks/__sql/oss_ci_benchmark_branches.sql
diff --git a/torchci/rockset/benchmarks/__sql/oss_ci_benchmark_llms.sql b/torchci/rockset/benchmarks/__sql/oss_ci_benchmark_llms.sql
diff --git a/torchci/rockset/benchmarks/__sql/oss_ci_benchmark_names.sql b/torchci/rockset/benchmarks/__sql/oss_ci_benchmark_names.sql
diff --git a/torchci/rockset/benchmarks/oss_ci_benchmark_branches.lambda.json b/torchci/rockset/benchmarks/oss_ci_benchmark_branches.lambda.json
diff --git a/torchci/rockset/benchmarks/oss_ci_benchmark_llms.lambda.json b/torchci/rockset/benchmarks/oss_ci_benchmark_llms.lambda.json
diff --git a/torchci/rockset/benchmarks/oss_ci_benchmark_names.lambda.json b/torchci/rockset/benchmarks/oss_ci_benchmark_names.lambda.json
diff --git a/torchci/rockset/commons/__sql/annotated_flaky_jobs.sql b/torchci/rockset/commons/__sql/annotated_flaky_jobs.sql