Skip to content

Commit 56b0a3c

Browse files
committed
Fix deprecation warnings.
1 parent cd453bc commit 56b0a3c

File tree

2 files changed

+15
-15
lines changed

2 files changed

+15
-15
lines changed

pymc3/sampling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -489,7 +489,7 @@ def _check_start_shape(model, start):
489489
# if start var has no shape
490490
else:
491491
# if model var has a specified shape
492-
if var_shape:
492+
if var_shape.size > 0:
493493
e += "\nExpected shape {} for var " \
494494
"'{}', got scalar {}".format(
495495
tuple(var_shape), var.name, start[var.name]
@@ -1523,4 +1523,4 @@ def init_nuts(init='auto', chains=1, n_init=500000, model=None,
15231523

15241524
step = pm.NUTS(potential=potential, model=model, **kwargs)
15251525

1526-
return start, step
1526+
return start, step

pymc3/tests/test_models_utils.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,15 @@ def assertMatrixLabels(self, m, l, mt=None, lt=None):
1313
assert np.all(
1414
np.equal(
1515
m.eval(),
16-
mt if mt is not None else self.data.as_matrix()
16+
mt if mt is not None else self.data.values
1717
)
1818
)
1919
assert l == list(lt or self.data.columns)
2020

2121
def test_numpy_init(self):
22-
m, l = utils.any_to_tensor_and_labels(self.data.as_matrix())
22+
m, l = utils.any_to_tensor_and_labels(self.data.values
2323
self.assertMatrixLabels(m, l, lt=['x0', 'x1'])
24-
m, l = utils.any_to_tensor_and_labels(self.data.as_matrix(), labels=['x2', 'x3'])
24+
m, l = utils.any_to_tensor_and_labels(self.data.values, labels=['x2', 'x3'])
2525
self.assertMatrixLabels(m, l, lt=['x2', 'x3'])
2626

2727
def test_pandas_init(self):
@@ -32,42 +32,42 @@ def test_pandas_init(self):
3232

3333
def test_dict_input(self):
3434
m, l = utils.any_to_tensor_and_labels(self.data.to_dict('dict'))
35-
self.assertMatrixLabels(m, l, mt=self.data.as_matrix(l), lt=l)
35+
self.assertMatrixLabels(m, l, mt=self.data[l].values, lt=l)
3636

3737
m, l = utils.any_to_tensor_and_labels(self.data.to_dict('series'))
38-
self.assertMatrixLabels(m, l, mt=self.data.as_matrix(l), lt=l)
38+
self.assertMatrixLabels(m, l, mt=self.data[l].values, lt=l)
3939

4040
m, l = utils.any_to_tensor_and_labels(self.data.to_dict('list'))
41-
self.assertMatrixLabels(m, l, mt=self.data.as_matrix(l), lt=l)
41+
self.assertMatrixLabels(m, l, mt=self.data[l].values, lt=l)
4242

4343
inp = {k: tt.as_tensor_variable(v) for k, v in self.data.to_dict('series').items()}
4444
m, l = utils.any_to_tensor_and_labels(inp)
45-
self.assertMatrixLabels(m, l, mt=self.data.as_matrix(l), lt=l)
45+
self.assertMatrixLabels(m, l, mt=self.data[l].values, lt=l)
4646

4747
def test_list_input(self):
48-
m, l = utils.any_to_tensor_and_labels(self.data.as_matrix().tolist())
48+
m, l = utils.any_to_tensor_and_labels(self.data.values.tolist())
4949
self.assertMatrixLabels(m, l, lt=['x0', 'x1'])
50-
m, l = utils.any_to_tensor_and_labels(self.data.as_matrix().tolist(), labels=['x2', 'x3'])
50+
m, l = utils.any_to_tensor_and_labels(self.data.values.tolist(), labels=['x2', 'x3'])
5151
self.assertMatrixLabels(m, l, lt=['x2', 'x3'])
5252

5353
def test_tensor_input(self):
5454
m, l = utils.any_to_tensor_and_labels(
55-
tt.as_tensor_variable(self.data.as_matrix().tolist()),
55+
tt.as_tensor_variable(self.data.values.tolist()),
5656
labels=['x0', 'x1']
5757
)
5858
self.assertMatrixLabels(m, l, lt=['x0', 'x1'])
5959
m, l = utils.any_to_tensor_and_labels(
60-
tt.as_tensor_variable(self.data.as_matrix().tolist()),
60+
tt.as_tensor_variable(self.data.values.tolist()),
6161
labels=['x2', 'x3'])
6262
self.assertMatrixLabels(m, l, lt=['x2', 'x3'])
6363

6464
def test_user_mistakes(self):
6565
# no labels for tensor variable
6666
with pytest.raises(
6767
ValueError):
68-
utils.any_to_tensor_and_labels(tt.as_tensor_variable(self.data.as_matrix().tolist()))
68+
utils.any_to_tensor_and_labels(tt.as_tensor_variable(self.data.values.tolist()))
6969
# len of labels is bad
7070
with pytest.raises(
7171
ValueError):
72-
utils.any_to_tensor_and_labels(self.data.as_matrix().tolist(),
72+
utils.any_to_tensor_and_labels(self.data.values.tolist(),
7373
labels=['x'])

0 commit comments

Comments
 (0)