Skip to content

Commit

Permalink
Replace pkg_resources.packaging (#3187)
Browse files Browse the repository at this point in the history
* testing pkg change

* removing packaging pin

* testing increased time

* testing docker change

* testing docker change

* adding disable token

* test

* Update test_container_model_prediction.sh
  • Loading branch information
udaij12 committed Jun 19, 2024
1 parent 9a3d785 commit ef9bcfe
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion requirements/common.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,4 @@ packaging==23.2
pynvml==11.5.0
pyyaml==6.0
ninja==1.11.1.1
setuptools==69.5.1
setuptools
2 changes: 1 addition & 1 deletion test/pytest/test_example_torch_compile.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@
import sys
from pathlib import Path

import packaging.version
import pytest
import torch
from pkg_resources import packaging

from ts.torch_handler.image_classifier import ImageClassifier
from ts.torch_handler.unit_tests.test_utils.mock_context import MockContext
Expand Down
2 changes: 1 addition & 1 deletion test/pytest/test_torch_compile.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
import time
from pathlib import Path

import packaging.version
import pytest
import torch
from pkg_resources import packaging
from test_data.torch_compile.compile_handler import CompileHandler

from ts.torch_handler.unit_tests.test_utils.mock_context import MockContext
Expand Down
2 changes: 1 addition & 1 deletion test/pytest/test_torch_export.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from pathlib import Path

import packaging.version
import torch
from pkg_resources import packaging

from ts.torch_handler.image_classifier import ImageClassifier
from ts.torch_handler.unit_tests.test_utils.mock_context import MockContext
Expand Down
2 changes: 1 addition & 1 deletion test/pytest/test_torch_xla.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@
import time
from pathlib import Path

import packaging.version
import pytest
from pkg_resources import packaging

try:
import torch_xla
Expand Down
2 changes: 1 addition & 1 deletion ts/torch_handler/base_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,8 @@
import os
import time

import packaging.version
import torch
from pkg_resources import packaging

from ts.handler_utils.timer import timed

Expand Down

0 comments on commit ef9bcfe

Please sign in to comment.