Skip to content

Commit cbb713f

Browse files
zaniebzanieb
authored
Review for #15769 (#15775)
Addresses my review comments from #15769
1 parent 2ec71bd commit cbb713f

File tree

1 file changed

+46
-45
lines changed

1 file changed

+46
-45
lines changed

‎crates/uv-torch/src/backend.rs‎

Lines changed: 46 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -713,7 +713,8 @@ impl TorchBackend {
713713
return None;
714714
}
715715
path_segments.next()?
716-
} else if index.host_str() == API_BASE_URL.strip_prefix("https://") {
716+
// TODO(zanieb): We should consolidate this with `is_known_url` somehow
717+
} else if index.host_str() == PYX_API_BASE_URL.strip_prefix("https://") {
717718
// E.g., `https://api.pyx.dev/simple/astral-sh/cu124`
718719
let mut path_segments = index.path_segments()?;
719720
if path_segments.next() != Some("simple") {
@@ -1110,180 +1111,180 @@ static PYTORCH_ROCM401_INDEX_URL: LazyLock<IndexUrl> =
11101111
static PYTORCH_XPU_INDEX_URL: LazyLock<IndexUrl> =
11111112
LazyLock::new(|| IndexUrl::from_str("https://download.pytorch.org/whl/xpu").unwrap());
11121113

1113-
static API_BASE_URL: LazyLock<Cow<'static, str>> = LazyLock::new(|| {
1114+
static PYX_API_BASE_URL: LazyLock<Cow<'static, str>> = LazyLock::new(|| {
11141115
std::env::var(EnvVars::PYX_API_URL)
11151116
.map(Cow::Owned)
11161117
.unwrap_or(Cow::Borrowed("https://api.pyx.dev"))
11171118
});
11181119
static PYX_CPU_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1119-
let api_base_url = &*API_BASE_URL;
1120+
let api_base_url = &*PYX_API_BASE_URL;
11201121
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cpu")).unwrap()
11211122
});
11221123
static PYX_CU129_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1123-
let api_base_url = &*API_BASE_URL;
1124+
let api_base_url = &*PYX_API_BASE_URL;
11241125
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu129")).unwrap()
11251126
});
11261127
static PYX_CU128_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1127-
let api_base_url = &*API_BASE_URL;
1128+
let api_base_url = &*PYX_API_BASE_URL;
11281129
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu128")).unwrap()
11291130
});
11301131
static PYX_CU126_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1131-
let api_base_url = &*API_BASE_URL;
1132+
let api_base_url = &*PYX_API_BASE_URL;
11321133
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu126")).unwrap()
11331134
});
11341135
static PYX_CU125_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1135-
let api_base_url = &*API_BASE_URL;
1136+
let api_base_url = &*PYX_API_BASE_URL;
11361137
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu125")).unwrap()
11371138
});
11381139
static PYX_CU124_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1139-
let api_base_url = &*API_BASE_URL;
1140+
let api_base_url = &*PYX_API_BASE_URL;
11401141
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu124")).unwrap()
11411142
});
11421143
static PYX_CU123_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1143-
let api_base_url = &*API_BASE_URL;
1144+
let api_base_url = &*PYX_API_BASE_URL;
11441145
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu123")).unwrap()
11451146
});
11461147
static PYX_CU122_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1147-
let api_base_url = &*API_BASE_URL;
1148+
let api_base_url = &*PYX_API_BASE_URL;
11481149
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu122")).unwrap()
11491150
});
11501151
static PYX_CU121_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1151-
let api_base_url = &*API_BASE_URL;
1152+
let api_base_url = &*PYX_API_BASE_URL;
11521153
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu121")).unwrap()
11531154
});
11541155
static PYX_CU120_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1155-
let api_base_url = &*API_BASE_URL;
1156+
let api_base_url = &*PYX_API_BASE_URL;
11561157
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu120")).unwrap()
11571158
});
11581159
static PYX_CU118_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1159-
let api_base_url = &*API_BASE_URL;
1160+
let api_base_url = &*PYX_API_BASE_URL;
11601161
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu118")).unwrap()
11611162
});
11621163
static PYX_CU117_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1163-
let api_base_url = &*API_BASE_URL;
1164+
let api_base_url = &*PYX_API_BASE_URL;
11641165
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu117")).unwrap()
11651166
});
11661167
static PYX_CU116_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1167-
let api_base_url = &*API_BASE_URL;
1168+
let api_base_url = &*PYX_API_BASE_URL;
11681169
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu116")).unwrap()
11691170
});
11701171
static PYX_CU115_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1171-
let api_base_url = &*API_BASE_URL;
1172+
let api_base_url = &*PYX_API_BASE_URL;
11721173
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu115")).unwrap()
11731174
});
11741175
static PYX_CU114_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1175-
let api_base_url = &*API_BASE_URL;
1176+
let api_base_url = &*PYX_API_BASE_URL;
11761177
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu114")).unwrap()
11771178
});
11781179
static PYX_CU113_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1179-
let api_base_url = &*API_BASE_URL;
1180+
let api_base_url = &*PYX_API_BASE_URL;
11801181
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu113")).unwrap()
11811182
});
11821183
static PYX_CU112_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1183-
let api_base_url = &*API_BASE_URL;
1184+
let api_base_url = &*PYX_API_BASE_URL;
11841185
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu112")).unwrap()
11851186
});
11861187
static PYX_CU111_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1187-
let api_base_url = &*API_BASE_URL;
1188+
let api_base_url = &*PYX_API_BASE_URL;
11881189
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu111")).unwrap()
11891190
});
11901191
static PYX_CU110_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1191-
let api_base_url = &*API_BASE_URL;
1192+
let api_base_url = &*PYX_API_BASE_URL;
11921193
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu110")).unwrap()
11931194
});
11941195
static PYX_CU102_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1195-
let api_base_url = &*API_BASE_URL;
1196+
let api_base_url = &*PYX_API_BASE_URL;
11961197
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu102")).unwrap()
11971198
});
11981199
static PYX_CU101_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1199-
let api_base_url = &*API_BASE_URL;
1200+
let api_base_url = &*PYX_API_BASE_URL;
12001201
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu101")).unwrap()
12011202
});
12021203
static PYX_CU100_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1203-
let api_base_url = &*API_BASE_URL;
1204+
let api_base_url = &*PYX_API_BASE_URL;
12041205
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu100")).unwrap()
12051206
});
12061207
static PYX_CU92_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1207-
let api_base_url = &*API_BASE_URL;
1208+
let api_base_url = &*PYX_API_BASE_URL;
12081209
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu92")).unwrap()
12091210
});
12101211
static PYX_CU91_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1211-
let api_base_url = &*API_BASE_URL;
1212+
let api_base_url = &*PYX_API_BASE_URL;
12121213
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu91")).unwrap()
12131214
});
12141215
static PYX_CU90_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1215-
let api_base_url = &*API_BASE_URL;
1216+
let api_base_url = &*PYX_API_BASE_URL;
12161217
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu90")).unwrap()
12171218
});
12181219
static PYX_CU80_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1219-
let api_base_url = &*API_BASE_URL;
1220+
let api_base_url = &*PYX_API_BASE_URL;
12201221
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/cu80")).unwrap()
12211222
});
12221223
static PYX_ROCM63_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1223-
let api_base_url = &*API_BASE_URL;
1224+
let api_base_url = &*PYX_API_BASE_URL;
12241225
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm6.3")).unwrap()
12251226
});
12261227
static PYX_ROCM624_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1227-
let api_base_url = &*API_BASE_URL;
1228+
let api_base_url = &*PYX_API_BASE_URL;
12281229
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm6.2.4")).unwrap()
12291230
});
12301231
static PYX_ROCM62_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1231-
let api_base_url = &*API_BASE_URL;
1232+
let api_base_url = &*PYX_API_BASE_URL;
12321233
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm6.2")).unwrap()
12331234
});
12341235
static PYX_ROCM61_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1235-
let api_base_url = &*API_BASE_URL;
1236+
let api_base_url = &*PYX_API_BASE_URL;
12361237
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm6.1")).unwrap()
12371238
});
12381239
static PYX_ROCM60_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1239-
let api_base_url = &*API_BASE_URL;
1240+
let api_base_url = &*PYX_API_BASE_URL;
12401241
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm6.0")).unwrap()
12411242
});
12421243
static PYX_ROCM57_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1243-
let api_base_url = &*API_BASE_URL;
1244+
let api_base_url = &*PYX_API_BASE_URL;
12441245
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.7")).unwrap()
12451246
});
12461247
static PYX_ROCM56_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1247-
let api_base_url = &*API_BASE_URL;
1248+
let api_base_url = &*PYX_API_BASE_URL;
12481249
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.6")).unwrap()
12491250
});
12501251
static PYX_ROCM55_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1251-
let api_base_url = &*API_BASE_URL;
1252+
let api_base_url = &*PYX_API_BASE_URL;
12521253
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.5")).unwrap()
12531254
});
12541255
static PYX_ROCM542_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1255-
let api_base_url = &*API_BASE_URL;
1256+
let api_base_url = &*PYX_API_BASE_URL;
12561257
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.4.2")).unwrap()
12571258
});
12581259
static PYX_ROCM54_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1259-
let api_base_url = &*API_BASE_URL;
1260+
let api_base_url = &*PYX_API_BASE_URL;
12601261
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.4")).unwrap()
12611262
});
12621263
static PYX_ROCM53_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1263-
let api_base_url = &*API_BASE_URL;
1264+
let api_base_url = &*PYX_API_BASE_URL;
12641265
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.3")).unwrap()
12651266
});
12661267
static PYX_ROCM52_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1267-
let api_base_url = &*API_BASE_URL;
1268+
let api_base_url = &*PYX_API_BASE_URL;
12681269
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.2")).unwrap()
12691270
});
12701271
static PYX_ROCM511_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1271-
let api_base_url = &*API_BASE_URL;
1272+
let api_base_url = &*PYX_API_BASE_URL;
12721273
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm5.1.1")).unwrap()
12731274
});
12741275
static PYX_ROCM42_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1275-
let api_base_url = &*API_BASE_URL;
1276+
let api_base_url = &*PYX_API_BASE_URL;
12761277
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm4.2")).unwrap()
12771278
});
12781279
static PYX_ROCM41_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1279-
let api_base_url = &*API_BASE_URL;
1280+
let api_base_url = &*PYX_API_BASE_URL;
12801281
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm4.1")).unwrap()
12811282
});
12821283
static PYX_ROCM401_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1283-
let api_base_url = &*API_BASE_URL;
1284+
let api_base_url = &*PYX_API_BASE_URL;
12841285
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/rocm4.0.1")).unwrap()
12851286
});
12861287
static PYX_XPU_INDEX_URL: LazyLock<IndexUrl> = LazyLock::new(|| {
1287-
let api_base_url = &*API_BASE_URL;
1288+
let api_base_url = &*PYX_API_BASE_URL;
12881289
IndexUrl::from_str(&format!("{api_base_url}/simple/astral-sh/xpu")).unwrap()
12891290
});

0 commit comments

Comments
 (0)