File tree Expand file tree Collapse file tree 3 files changed +8
-8
lines changed Expand file tree Collapse file tree 3 files changed +8
-8
lines changed Original file line number Diff line number Diff line change 23
23
fail-fast : false
24
24
matrix :
25
25
include :
26
- - os : linux.2xlarge
27
- python-version : 3.8
28
- python-tag : " py38"
29
26
- os : linux.2xlarge
30
27
python-version : 3.9
31
28
python-tag : " py39"
35
32
- os : linux.2xlarge
36
33
python-version : ' 3.11'
37
34
python-tag : " py311"
35
+ - os : linux.2xlarge
36
+ python-version : ' 3.12'
37
+ python-tag : " py312"
38
38
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
39
39
with :
40
40
runner : ${{ matrix.os }}
Original file line number Diff line number Diff line change @@ -257,11 +257,11 @@ def test_warmup_checkpointing(self) -> None:
257
257
labels = {"DefaultTask" : model_output ["label" ]},
258
258
weights = {"DefaultTask" : model_output ["weight" ]},
259
259
)
260
- self .assertEquals (
260
+ self .assertEqual (
261
261
qps ._metrics_computations [0 ].warmup_examples ,
262
262
batch_size * warmup_steps * (i + 1 ),
263
263
)
264
- self .assertEquals (
264
+ self .assertEqual (
265
265
qps ._metrics_computations [0 ].num_examples ,
266
266
batch_size * (warmup_steps + extra_steps ) * (i + 1 ),
267
267
)
@@ -303,9 +303,9 @@ def test_mtml_empty_update(self) -> None:
303
303
else :
304
304
del labels ["t2" ]
305
305
qps .update (predictions = predictions , labels = labels , weights = weights )
306
- self .assertEquals (
306
+ self .assertEqual (
307
307
qps ._metrics_computations [0 ].num_examples , (step + 1 ) // 2 * batch_size
308
308
)
309
- self .assertEquals (
309
+ self .assertEqual (
310
310
qps ._metrics_computations [1 ].num_examples , (step + 2 ) // 2 * batch_size
311
311
)
Original file line number Diff line number Diff line change @@ -349,7 +349,7 @@ def test_device(self) -> None:
349
349
tables = [config ],
350
350
device = torch .device ("meta" ),
351
351
)
352
- self .assertEquals (torch .device ("meta" ), ec .embeddings ["t1" ].weight .device )
352
+ self .assertEqual (torch .device ("meta" ), ec .embeddings ["t1" ].weight .device )
353
353
354
354
def test_duplicate_config_name_fails (self ) -> None :
355
355
e1_config = EmbeddingConfig (
You can’t perform that action at this time.
0 commit comments