4331fd73f0864d21c366a24c0105c8bd3555189b,tests/test_networkml.py,,test_networkml_test_randomforest,#,93

Before Change

def test_networkml_test_randomforest():
    os.environ["POSEIDON_PUBLIC_SESSIONS"] = ""
    tempdir = tempfile.TemporaryDirectory()
    savew = os.path.join(tempdir.name, "not.used")
    netml = run_networkml(
        ["-p", "tests/", "-o", "test", "-a", "randomforest"],
        savew=savew,
        model="networkml/trained_models/randomforest/RandomForestModel.pkl")

After Change

def test_networkml_test_randomforest():
    os.environ["POSEIDON_PUBLIC_SESSIONS"] = ""
    with tempfile.TemporaryDirectory() as tempdir:
        savew = os.path.join(tempdir, "not.used")
        netml = run_networkml(
            ["-p", "tests/", "-o", "test", "-a", "randomforest"],
            savew=savew,
            model="networkml/trained_models/randomforest/RandomForestModel.pkl")
        assert netml.model.feature_list

Instances