DataRobot Python API Upload Dataset from Catalog

upload_dataset_from_catalog.py

def upload_dataset_from_catalog(self, dataset):
    import json
    import time
    client = dr.Client()
    res = client.post(client.endpoint + "/projects/{}/predictionDatasets/datasetUploads/".format(self.id), json={"datasetId": dataset.id})
    dataset_id = json.loads(res.text)['datasetId']
    while dataset_id not in [ds.id for ds in self.get_datasets()]:
        time.sleep(1)
    return dataset_id
dr.Project.upload_dataset_from_catalog = upload_dataset_from_catalog