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
@@ -50,7 +50,6 @@ from transformers.testing_utils import (
|
||||
RequestCounter,
|
||||
is_staging_test,
|
||||
nested_simplify,
|
||||
require_scatter,
|
||||
require_tensorflow_probability,
|
||||
require_tf,
|
||||
require_torch,
|
||||
@@ -749,7 +748,6 @@ class PipelineUtilsTest(unittest.TestCase):
|
||||
|
||||
@slow
|
||||
@require_torch
|
||||
@require_scatter
|
||||
def test_load_default_pipelines_pt_table_qa(self):
|
||||
import torch
|
||||
|
||||
|
||||
Reference in New Issue
Block a user