From 32ee3b393038b112890c01690e3ea3d7e75a837e Mon Sep 17 00:00:00 2001
From: Oliver Borchert <borchero@icloud.com>
Date: Sun, 29 Mar 2020 04:30:17 +0200
Subject: [PATCH] Update dependencies

---
 pycave/bayes/gmm.py    | 8 ++++----
 pycave/bayes/markov.py | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/pycave/bayes/gmm.py b/pycave/bayes/gmm.py
index 4c81da8..3b45f6b 100644
--- a/pycave/bayes/gmm.py
+++ b/pycave/bayes/gmm.py
@@ -3,9 +3,9 @@
 import torch
 import torch.nn as nn
 import torch.distributions as dist
-import bxtorch.nn as xnn
-from bxtorch.utils.stdio import ProgressBar
-from bxtorch.utils.torch import to_one_hot
+import pyblaze.nn as xnn
+from pyblaze.utils.stdio import ProgressBar
+from pyblaze.utils.torch import to_one_hot
 from sklearn.cluster import KMeans
 from .utils import log_normal, log_responsibilities, max_likeli_means, max_likeli_covars
 
@@ -166,7 +166,7 @@ def fit(self, data, max_iter=100, batch_size=None, eps=1e-7, verbose=False):
 
         Returns
         -------
-        bxtorch.nn.History
+        pyblaze.nn.History
             A history object containing the negative log-likelihoods over the course of the
             EM-training.
         """
diff --git a/pycave/bayes/markov.py b/pycave/bayes/markov.py
index 6c4b0d4..6e48ba4 100644
--- a/pycave/bayes/markov.py
+++ b/pycave/bayes/markov.py
@@ -2,7 +2,7 @@
 import torch
 import torch.nn as nn
 import torch.distributions as dist
-import bxtorch.nn as xnn
+import pyblaze.nn as xnn
 from .utils import power_iteration
 
 class MarkovModelConfig(xnn.Config):
@@ -77,7 +77,7 @@ def fit(self, sequences, teleport_alpha=0):
 
         Returns
         -------
-        bxtorch.nn.History
+        pyblaze.nn.History
             For completeness, it returns a history object. However, apart from the duration of the
             training, the object does not carry any information.
         """