900c3874bb233c6995fb40a86a0d30a7ae177a41,tests/distributions/test_pickle.py,,test_pickle,#Any#,59

Before Change

    // Note that pickling torch.Size() requires protocol >= 2
    serialized = pickle.dumps(dist, pickle.HIGHEST_PROTOCOL)
    deserialized = pickle.loads(serialized)
    assert isinstance(deserialized, Dist)

After Change

    except Exception:
        pytest.skip(msg="cannot construct distribution")

    buffer = io.BytesIO()
    // Note that pickling torch.Size() requires protocol >= 2
    torch.save(dist, buffer, pickle_protocol=pickle.HIGHEST_PROTOCOL)
    buffer.seek(0)
    deserialized = torch.load(buffer)

Instances