fix lint/mypy failures (#7183)

* fix lint check * fix mypy
dagster-io · Mar 24, 2022 · c1d674d · c1d674d
1 parent 1fff21f
commit c1d674d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/python_modules/dagster/dagster/core/definitions/job_definition.py b/python_modules/dagster/dagster/core/definitions/job_definition.py
@@ -186,6 +186,11 @@ def execute_in_process(
                 "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)
             tags = partition_set.tags_for_partition(partition)

diff --git a/python_modules/dagster/dagster/core/definitions/partition.py b/python_modules/dagster/dagster/core/definitions/partition.py
@@ -9,7 +9,6 @@
 from dateutil.relativedelta import relativedelta
 
 from dagster import check
-from dagster.core.storage.tags import PARTITION_NAME_TAG
 from dagster.serdes import whitelist_for_serdes
 
 from ...seven.compat.pendulum import PendulumDateTime, to_timezone