Searched +full:rockset +full:- +full:api +full:- +full:key (Results 1 – 7 of 7) sorted by relevance
11 import rockset # type: ignore[import]16 def get_oncall_from_testfile(testfile: str) -> list[str] | None:42 def get_test_stat_aggregates(date: datetime.date) -> Any:43 # Initialize the Rockset client with your API key45 rockset_api_server = "api.rs2.usw2.rockset.com"47 rs = rockset.RocksetClient(host="api.usw2a1.rockset.com", api_key=rockset_api_key)49 # Define the name of the Rockset collection and lambda function53 rockset.models.QueryParameter(name="startTime", type="string", value=iso_date)70 description="Upload test stat aggregates to Rockset."73 "--date",[all …]
14 import rockset # type: ignore[import]17 PYTORCH_REPO = "https://api.github.com/repos/pytorch/pytorch"23 # NB: Rockset has an upper limit of 5000 documents in one request27 def _get_request_headers() -> dict[str, str]:34 def _get_artifact_urls(prefix: str, workflow_run_id: int) -> dict[Path, str]:35 """Get all workflow artifacts with 'test-report' in the name."""56 ) -> Path:59 # re-run a workflow and produce a new set of artifacts. To avoid name60 # collisions, we add `-runattempt1<run #>-` somewhere in the artifact name.64 atoms = str(artifact_name).split("-")[all …]
18 "facebook-github-bot",19 "pytorch-bot[bot]",32 ) -> Any:34 if token is not None and url.startswith("https://api.github.com/"):44 key in err.headers for key in ["X-RateLimit-Limit", "X-RateLimit-Used"]47 … f"Rate limit exceeded: {err.headers['X-RateLimit-Used']}/{err.headers['X-RateLimit-Limit']}"56 ) -> list[dict[str, Any]]:70 ) -> list[dict[str, Any]]:77 period_end_date = period_begin_date + datetime.timedelta(days=period_length - 1)84 "https://api.github.com/search/issues",[all …]
5 - cron: 17,47 * * * *10 cancel-in-progress: false15 runs-on: ubuntu-20.0418 - name: Update viable/strict19 uses: pytorch/test-infra/.github/actions/update-viablestrict@release/2.422 stable-branch: viable/strict23 requires: '[\"pull\", \"trunk\", \"lint\", \"linux-binary\"]'24 secret-bot-token: ${{ secrets.MERGEBOT_TOKEN }}25 rockset-api-key: ${{ secrets.ROCKSET_API_KEY }}
8 # And re-run the test locally with ones PAT54 ) -> Any:57 def get_mocked_queries() -> Any:60 with gzip.open(gql_db_fname, encoding="utf-8", mode="rt") as f:63 def save_mocked_queries(obj: Any) -> None:64 with gzip.open(gql_db_fname, encoding="utf-8", mode="wt") as f:68 key = key_function(*args)71 if key in mocked_queries:72 return mocked_queries[key]81 err_msg += " the rockset api key passed via ROCKSET_API_KEY,"[all …]
... xla hash.", 13 "headRefName": "update-xla-commit-hash/5573005593-54-1", ...