Allow torch>=2 (#3164)

pyro-ppl · Dec 12, 2022 · 77a67ff · 77a67ff
1 parent 6d61ea9
commit 77a67ff
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/pyro/distributions/torch_patch.py b/pyro/distributions/torch_patch.py
@@ -7,8 +7,6 @@
 
 import torch
 
-assert torch.__version__.startswith("1.")
-
 
 def patch_dependency(target, root_module=torch):
     parts = target.split(".")