Skip to content

Commit

Permalink
Fix torch GPU CI
Browse files Browse the repository at this point in the history
  • Loading branch information
fchollet committed Jan 2, 2025
1 parent 5b29974 commit 2c95bfe
Showing 1 changed file with 1 addition and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -183,9 +183,7 @@ def test_random_shear_tf_data_bounding_boxes(
)

transformation = {
"shear_factor": backend_utils.convert_tf_tensor(
np.array(translation)
),
"shear_factor": np.array(translation),
"input_shape": image_shape,
}

Expand Down

0 comments on commit 2c95bfe

Please sign in to comment.