Skip to content

Commit 4364e9a

Browse files
committed
Drop 3.8 CI and add 3.12 (#2296)
Summary: Pull Request resolved: #2296 Reviewed By: dstaay-fb, gnahzg Differential Revision: D61221869 fbshipit-source-id: 95d87098dea6c8cdf67cd3202b4946c73c2f32ab
1 parent 910f69c commit 4364e9a

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

.github/workflows/unittest_ci_cpu.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,6 @@ jobs:
2323
fail-fast: false
2424
matrix:
2525
include:
26-
- os: linux.2xlarge
27-
python-version: 3.8
28-
python-tag: "py38"
2926
- os: linux.2xlarge
3027
python-version: 3.9
3128
python-tag: "py39"
@@ -35,6 +32,9 @@ jobs:
3532
- os: linux.2xlarge
3633
python-version: '3.11'
3734
python-tag: "py311"
35+
- os: linux.2xlarge
36+
python-version: '3.12'
37+
python-tag: "py312"
3838
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
3939
with:
4040
runner: ${{ matrix.os }}

torchrec/metrics/tests/test_tower_qps.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -257,11 +257,11 @@ def test_warmup_checkpointing(self) -> None:
257257
labels={"DefaultTask": model_output["label"]},
258258
weights={"DefaultTask": model_output["weight"]},
259259
)
260-
self.assertEquals(
260+
self.assertEqual(
261261
qps._metrics_computations[0].warmup_examples,
262262
batch_size * warmup_steps * (i + 1),
263263
)
264-
self.assertEquals(
264+
self.assertEqual(
265265
qps._metrics_computations[0].num_examples,
266266
batch_size * (warmup_steps + extra_steps) * (i + 1),
267267
)
@@ -303,9 +303,9 @@ def test_mtml_empty_update(self) -> None:
303303
else:
304304
del labels["t2"]
305305
qps.update(predictions=predictions, labels=labels, weights=weights)
306-
self.assertEquals(
306+
self.assertEqual(
307307
qps._metrics_computations[0].num_examples, (step + 1) // 2 * batch_size
308308
)
309-
self.assertEquals(
309+
self.assertEqual(
310310
qps._metrics_computations[1].num_examples, (step + 2) // 2 * batch_size
311311
)

torchrec/modules/tests/test_embedding_modules.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ def test_device(self) -> None:
349349
tables=[config],
350350
device=torch.device("meta"),
351351
)
352-
self.assertEquals(torch.device("meta"), ec.embeddings["t1"].weight.device)
352+
self.assertEqual(torch.device("meta"), ec.embeddings["t1"].weight.device)
353353

354354
def test_duplicate_config_name_fails(self) -> None:
355355
e1_config = EmbeddingConfig(

0 commit comments

Comments
 (0)