Modify Foundry Schema in Code

Heyo :wave:. Question!

Is there a way to modify the JSON Foundry Schema of a dataset from the pyspark code? Specifically looking to change the primaryKey field from null to

"primaryKey": {
    "columns": [
      "id"
    ],
    "resolution": {
      "type": "duplicate",
      "duplicate": {
        "resolutionStrategy": {
          "type": "latestWins",
          "latestWins": {
            "orderingColumn": "ingest_timestamp"
          }
        },
        "deletionColumn": null
      }
    }
  }

in the code instead of manually pasting into the schema ui