Unit Test Fails (2): test_infectionsrtfeedback.py

[akruel01]

Line 87: samp1 = InfectionsWithFeedback.sample(
            gen_int=gen_int,
            Rt=Rt,
            I0=I0,
        )
pyrenew/latent/infectionswithfeedback.py:116 > if inf_feedback_strength.size(0) == 1:
AttributeError: 'tuple' object has no attribute 'size'

self = <pyrenew.latent.infectionswithfeedback.InfectionsWithFeedback object at 0x145678f80>
Rt = tensor([0.5000, 0.6000, 0.7000, 0.8000, 2.0000, 0.5000, 2.2500])
I0 = tensor([0., 0., 0., 0., 0., 0., 1.])
gen_int = tensor([0.4000, 0.2500, 0.2500, 0.1000, 0.0000, 0.0000, 0.0000]), kwargs = {}
gen_int_rev = tensor([0.0000, 0.0000, 0.0000, 0.1000, 0.2500, 0.2500, 0.4000])
recent_I0 = tensor([0., 0., 0., 0., 0., 0., 1.])
inf_feedback_strength = (tensor([0., 0., 0., 0., 0., 0., 0.]),)

---

Line 128: samp1 = InfectionsWithFeedback.sample(
            gen_int=gen_int,
            Rt=Rt,
            I0=I0,
        )
pyrenew/latent/infectionswithfeedback.py:116 > if inf_feedback_strength.size(0) == 1:

self = <pyrenew.latent.infectionswithfeedback.InfectionsWithFeedback object at 0x148c5bdd0>
Rt = tensor([0.5000, 0.6000, 1.5000, 2.5230, 0.7000, 0.8000])
I0 = tensor([0., 0., 0., 0., 0., 0., 1.])
gen_int = tensor([0.4000, 0.2500, 0.2500, 0.1000, 0.0000, 0.0000, 0.0000]), kwargs = {}
gen_int_rev = tensor([0.0000, 0.0000, 0.0000, 0.1000, 0.2500, 0.2500, 0.4000])
recent_I0 = tensor([0., 0., 0., 0., 0., 0., 1.])
inf_feedback_strength = (tensor([0.5000, 0.5000, 0.5000, 0.5000, 0.5000, 0.5000]),)