fix: wrong order in variable delarations and linting errors;

WenjieDu · Nov 18, 2023 · f80467d · f80467d
1 parent f6d4e37
commit f80467d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/pypots/clustering/crli/model.py b/pypots/clustering/crli/model.py
@@ -11,7 +11,7 @@
 # License: BSD-3-Clause
 
 import os
-from typing import Union, Optional, Tuple
+from typing import Union, Optional
 
 import numpy as np
 import torch

diff --git a/pypots/clustering/vader/model.py b/pypots/clustering/vader/model.py
@@ -12,7 +12,7 @@
 
 
 import os
-from typing import Tuple, Union, Optional
+from typing import Union, Optional
 
 import numpy as np
 import torch
@@ -443,6 +443,15 @@ def predict(
                 inputs = self._assemble_input_for_testing(data)
                 results = self.model.forward(inputs, training=False)
 
+                mu_tilde = results["mu_tilde"].cpu().numpy()
+                mu_tilde_collector.append(mu_tilde)
+                mu = results["mu"].cpu().numpy()
+                mu_collector.append(mu)
+                var = results["var"].cpu().numpy()
+                var_collector.append(var)
+                phi = results["phi"].cpu().numpy()
+                phi_collector.append(phi)
+
                 def func_to_apply(
                     mu_t_: np.ndarray,
                     mu_: np.ndarray,
@@ -465,16 +474,8 @@ def func_to_apply(
                 clustering_results_collector.append(clustering_results)
 
                 if return_latent_vars:
-                    mu_tilde = results["mu_tilde"].cpu().numpy()
-                    mu_tilde_collector.append(mu_tilde)
                     stddev_tilde = results["stddev_tilde"].cpu().numpy()
                     stddev_tilde_collector.append(stddev_tilde)
-                    mu = results["mu"].cpu().numpy()
-                    mu_collector.append(mu)
-                    var = results["var"].cpu().numpy()
-                    var_collector.append(var)
-                    phi = results["phi"].cpu().numpy()
-                    phi_collector.append(phi)
                     z = results["z"].cpu().numpy()
                     z_collector.append(z)
                     imputation_latent = results["imputation_latent"].cpu().numpy()