wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-09-10 12:37:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix for pytorch 1.5 and pydantic

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-04-21 22:27:05 +02:00
parent c5fac32bb2
commit 8a0e59e47c
2 changed files with 9 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/modules/test_distribution_output.py
+3 -3
View File
@@ -230,9 +230,9 @@ def test_lowrank_multivariate_normal() -> None:
Sigma = cov_factor @ cov_factor.T + cov_diag
distr = LowRankMultivariateNormal(
loc=torch.Tensor(loc),
cov_diag=torch.Tensor(np.diag(cov_diag)),
cov_factor=torch.Tensor(cov_factor),
loc=torch.Tensor(loc.copy()),
cov_diag=torch.Tensor(np.diag(cov_diag).copy()),
cov_factor=torch.Tensor(cov_factor.copy()),
)
assert np.allclose(