Skip to content

Commit

Permalink
DO NOT MERGE: Temporarily removing tags to see if we are not relying …
Browse files Browse the repository at this point in the history
…on them anymore
  • Loading branch information
ricardoV94 committed Nov 10, 2022
1 parent 784eee0 commit 0c5d5f8
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions pymc/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -1315,7 +1315,7 @@ def register_rv(
name = self.name_for(name)
rv_var.name = name
# TODO: Stop adding mapping information to the tag
rv_var.tag.total_size = total_size
# rv_var.tag.total_size = total_size
self.rvs_to_total_size[rv_var] = total_size

# Associate previously unknown dimension names with
Expand Down Expand Up @@ -1453,7 +1453,7 @@ def make_obs_var(
_, size, _, *inps = observed_rv_var.owner.inputs
observed_rv_var = observed_rv_var.owner.op(*inps, size=size, name=f"{name}_observed")
# TODO: Stop adding mapping information to the tag
observed_rv_var.tag.observations = nonmissing_data
# observed_rv_var.tag.observations = nonmissing_data

self.create_value_var(observed_rv_var, transform=None, value_var=nonmissing_data)
self.add_random_variable(observed_rv_var)
Expand All @@ -1471,7 +1471,7 @@ def make_obs_var(
else:
data = at.as_tensor_variable(data, name=name)
# TODO: Stop adding mapping information to the tag
rv_var.tag.observations = data
# rv_var.tag.observations = data
self.create_value_var(rv_var, transform=None, value_var=data)
self.add_random_variable(rv_var, dims)
self.observed_RVs.append(rv_var)
Expand Down Expand Up @@ -1499,7 +1499,7 @@ def create_value_var(
value_var.tag.test_value = rv_var.tag.test_value

# TODO: Stop adding mapping information to the tag
rv_var.tag.value_var = value_var
# rv_var.tag.value_var = value_var

# Make the value variable a transformed value variable,
# if there's an applicable transform
Expand All @@ -1508,7 +1508,7 @@ def create_value_var(

if transform is not None and transform is not UNSET:
# TODO: Stop adding mapping information to the tag
value_var.tag.transform = transform
# value_var.tag.transform = transform
value_var.name = f"{value_var.name}_{transform.name}__"
if aesara.config.compute_test_value != "off":
value_var.tag.test_value = transform.forward(
Expand Down

0 comments on commit 0c5d5f8

Please sign in to comment.