Remove unused imports

mit-han-lab · May 31, 2024 · 428be4a · 428be4a
1 parent b0d8446
commit 428be4a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/examples/QCBM/qcbm_gaussian_mixture.py b/examples/QCBM/qcbm_gaussian_mixture.py
@@ -1,7 +1,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import torch
-import torch.nn as nn
 from torchquantum.algorithm import QCBM, MMDLoss
 import torchquantum as tq
 

diff --git a/test/algorithm/test_qcbm.py b/test/algorithm/test_qcbm.py
@@ -1,7 +1,6 @@
 from torchquantum.algorithm.qcbm import QCBM, MMDLoss
 import torchquantum as tq
 import torch
-import pytest
 
 
 def test_qcbm_forward():
@@ -29,5 +28,4 @@ def test_mmd_loss():
 
     mmd = MMDLoss(bandwidth, space)
     loss = mmd(torch.zeros(4), torch.zeros(4))
-    print(loss)
     assert torch.isclose(loss, torch.tensor(0.0), rtol=1e-5)