From 21371596735ec86f0d20a861094733a17732279a Mon Sep 17 00:00:00 2001
From: "Mikkel N. Schmidt" <mnsc@dtu.dk>
Date: Wed, 27 Apr 2022 17:13:17 +0200
Subject: [PATCH] demo bug fix, plotting works
---
demos/regression.py | 2 +-
supr/utils.py | 3 ++-
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/demos/regression.py b/demos/regression.py
index 48d477c..b6e9e09 100644
--- a/demos/regression.py
+++ b/demos/regression.py
@@ -3,7 +3,7 @@ import numpy as np
import matplotlib.pyplot as plt
import torch
import supr
-from tspn.utils import drawnow, arrange_figs
+from supr.utils import drawnow
from scipy.stats import norm
#%% Dataset
diff --git a/supr/utils.py b/supr/utils.py
index 790caf1..3ee9c1f 100644
--- a/supr/utils.py
+++ b/supr/utils.py
@@ -9,6 +9,7 @@ from PyQt5 import QtWidgets
# %%
def drawnow():
+ plt.show(block=False)
plt.gcf().canvas.draw()
plt.gcf().canvas.flush_events()
@@ -67,4 +68,4 @@ def local_scramble_2d(dist: float, dim: tuple):
def bsgen(v, v0):
while v > v0:
yield v
- v = math.ceil(v / 2)
\ No newline at end of file
+ v = math.ceil(v / 2)
--
GitLab