kg_emb/ references SampleBaseDataset which appears to have been renamed to SampleDataset in PyHealth 2.0. Anyone who imports pyhealth.medcode.pretrained_embeddings gets an ImportError and package would crash immediately. Found this while implementing KEEP embeddings in personal fork. Currently adding a try/e bandaid in my fork but is out of scope for the KEEP work. Likely impacts all 6 files in kg_emb.models and the 2 in kg_emb/datasets. I can rename SampleBaseDataset --> SampleDataset across the 8 files if no one gets to it.
kg_emb/ references
SampleBaseDatasetwhich appears to have been renamed toSampleDatasetin PyHealth 2.0. Anyone who importspyhealth.medcode.pretrained_embeddingsgets an ImportError and package would crash immediately. Found this while implementing KEEP embeddings in personal fork. Currently adding a try/e bandaid in my fork but is out of scope for the KEEP work. Likely impacts all 6 files in kg_emb.models and the 2 in kg_emb/datasets. I can renameSampleBaseDataset-->SampleDatasetacross the 8 files if no one gets to it.