Facebook
From Insensitive Horse, 11 Months ago, written in Python.
Embed
Download Paste or View Raw
Hits: 84
  1. from dagster import Config
  2. from dagster_graphql import DagsterGraphQLClient, DagsterGraphQLClientError
  3.  
  4.  
  5. class MyAssetConfig(Config):
  6.     date_col_name: str
  7.     pred_col_name: str
  8.     steps_to_predict: int
  9.     lags: int
  10.  
  11.  
  12. def execute_job():
  13.     url = "https://my_cloud/prod"
  14.     token = os.environ["USER_TOKEN"]
  15.     client = DagsterGraphQLClient(
  16.         url,
  17.         transport=RequestsHTTPTransport(
  18.             url=url + "/graphql", headers={"Dagster-Cloud-Api-Token": token}
  19.         ),
  20.     )
  21.  
  22.     try:
  23.         new_run_id: str = client.submit_job_execution(
  24.             job_name="ts_predict",
  25.             repository_location_name="whatever",
  26.             repository_name="__repository__",  # by default
  27.             run_config={"ops"= "timeseries_predict": MyAssetConfig(date_col_name="Date", pred_col_name="bla",
  28.                                                                  steps_to_predict=10, lags=10)}
  29.         )
  30.         print("executed")
  31.         return "<p>Job executed!</p>"
  32.     except DagsterGraphQLClientError as exc:
  33.         print(exc)
  34.         return f"<p>{exc}</p>"