Update distributions.py

NikitinaMaria · web-flow · commit 1a1192a5c3e4 · 2024-11-04T20:31:35.000+03:00
diff --git a/src/irt/distributions.py b/src/irt/distributions.py
@@ -1,3 +1,5 @@
+import torch
+
 class torch.distributions.Distribution:
 	'''
 	The abstract base class for probability distributions, which we inherit from. These methods are implied

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+import torch`
	`2`	`+`
`1`	`3`	`class torch.distributions.Distribution:`
`2`	`4`	`'''`
`3`	`5`	`The abstract base class for probability distributions, which we inherit from. These methods are implied`