Fix tapas scatter (#20149)
* First draft * Remove scatter dependency * Add require_torch * update vectorized sum test, add clone call * remove artifacts * fix style * fix style v2 * remove "scatter" mentions from the code base * fix isort error Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local> Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
committed by
GitHub
parent
f711d683b5
commit
78a471ff71
@@ -65,7 +65,6 @@ from .utils import (
|
||||
is_pytorch_quantization_available,
|
||||
is_rjieba_available,
|
||||
is_safetensors_available,
|
||||
is_scatter_available,
|
||||
is_scipy_available,
|
||||
is_sentencepiece_available,
|
||||
is_soundfile_availble,
|
||||
@@ -319,16 +318,6 @@ def require_intel_extension_for_pytorch(test_case):
|
||||
)(test_case)
|
||||
|
||||
|
||||
def require_torch_scatter(test_case):
|
||||
"""
|
||||
Decorator marking a test that requires PyTorch scatter.
|
||||
|
||||
These tests are skipped when PyTorch scatter isn't installed.
|
||||
|
||||
"""
|
||||
return unittest.skipUnless(is_scatter_available(), "test requires PyTorch scatter")(test_case)
|
||||
|
||||
|
||||
def require_tensorflow_probability(test_case):
|
||||
"""
|
||||
Decorator marking a test that requires TensorFlow probability.
|
||||
@@ -405,14 +394,6 @@ def require_pytesseract(test_case):
|
||||
return unittest.skipUnless(is_pytesseract_available(), "test requires PyTesseract")(test_case)
|
||||
|
||||
|
||||
def require_scatter(test_case):
|
||||
"""
|
||||
Decorator marking a test that requires PyTorch Scatter. These tests are skipped when PyTorch Scatter isn't
|
||||
installed.
|
||||
"""
|
||||
return unittest.skipUnless(is_scatter_available(), "test requires PyTorch Scatter")(test_case)
|
||||
|
||||
|
||||
def require_pytorch_quantization(test_case):
|
||||
"""
|
||||
Decorator marking a test that requires PyTorch Quantization Toolkit. These tests are skipped when PyTorch
|
||||
|
||||
Reference in New Issue
Block a user