Add back OSError handling that was removed in 48e1ba8

eecs485staff · Nov 8, 2024 · 1d4e8ab · 1d4e8ab
1 parent 5294aa6
commit 1d4e8ab
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}"