Merge pull request #80 from eecs485staff/add-back-OSError

Catch more `OSError`s
eecs485staff · Nov 8, 2024 · 042f54b · 042f54b
2 parents 5294aa6 + ac60215
commit 042f54b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/madoop/mapreduce.py b/madoop/mapreduce.py
@@ -199,7 +199,7 @@ def map_single_chunk(exe, input_path, output_path, chunk):
                 input=chunk,
                 stdout=outfile,
             )
-        except subprocess.CalledProcessError as err:
+        except (subprocess.CalledProcessError, OSError) as err:
             raise MadoopError(
                 f"Command returned non-zero: "
                 f"{exe} < {input_path} > {output_path}"
@@ -420,7 +420,7 @@ def reduce_single_file(exe, input_path, output_path):
                 stdin=infile,
                 stdout=outfile,
             )
-        except subprocess.CalledProcessError as err:
+        except (subprocess.CalledProcessError, OSError) as err:
             raise MadoopError(
                 f"Command returned non-zero: "
                 f"{exe} < {input_path} > {output_path}"