Add test for functorch

pytorch-labs · Mar 14, 2024 · 0103fc0 · 0103fc0
1 parent 1432845
commit 0103fc0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/fixtures/deprecated_symbols/checker/functorch.py b/tests/fixtures/deprecated_symbols/checker/functorch.py
@@ -2,3 +2,5 @@
 
 # Check that we get only one warning for the line
 functorch.vmap(tdmodule, (None, 0))(td, params)
+
+from functorch import vmap, jacrev
diff --git a/tests/fixtures/deprecated_symbols/checker/functorch.txt b/tests/fixtures/deprecated_symbols/checker/functorch.txt
@@ -1 +1,3 @@
 4:1 TOR101 Use of deprecated function functorch.vmap
+6:1 TOR103 Import of deprecated function functorch.vmap
+6:1 TOR103 Import of deprecated function functorch.jacrev
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,3 +2,5 @@

		# Check that we get only one warning for the line
		functorch.vmap(tdmodule, (None, 0))(td, params)

		from functorch import vmap, jacrev