Support overriding run config for partitions with execute_in_process (#…

…10246)

Fixes #10183

python_modules/dagster/dagster/_core/definitions/job_definition.py

@@ -383,18 +383,16 @@ def execute_in_process(
                 check.failed(
                     f"Provided partition key `{partition_key}` for job `{self._name}` without a partitioned config"
                 )
-            check.invariant(
-                not run_config,
-                "Cannot provide both run_config and partition_key arguments to `execute_in_process`",
-            )
             partition_set = self.get_partition_set_def()
             if not partition_set:
                 check.failed(
                     f"Provided partition key `{partition_key}` for job `{self._name}` without a partitioned config"
                 )
 
             partition = partition_set.get_partition(partition_key)
-            run_config = partition_set.run_config_for_partition(partition)
+            run_config = (
+                run_config if run_config else partition_set.run_config_for_partition(partition)
+            )
             tags = partition_set.tags_for_partition(partition)
 
         return core_execute_in_process(

python_modules/dagster/dagster_tests/core_tests/asset_defs_tests/test_materialize_to_memory.py

@@ -237,6 +237,22 @@ def the_asset(context):
     assert result.success
 
 
+def test_materialize_to_memory_partition_key_and_run_config():
+    @asset(config_schema={"value": str})
+    def configurable(context):
+        assert context.op_config["value"] == "a"
+
+    @asset(partitions_def=DailyPartitionsDefinition(start_date="2022-09-11"))
+    def partitioned(context):
+        assert context.partition_key == "2022-09-11"
+
+    assert materialize_to_memory(
+        [partitioned, configurable],
+        partition_key="2022-09-11",
+        run_config={"ops": {"configurable": {"config": {"value": "a"}}}},
+    ).success
+
+
 def test_materialize_to_memory_provided_io_manager_instance():
     @io_manager
     def the_manager():