From dc22a954a2a286a1356b80f37fefdc0c5b18b1eb Mon Sep 17 00:00:00 2001
From: Daniel Olsen <daniel.olsen@breakthroughenergy.org>
Date: Thu, 1 Oct 2020 13:38:57 -0700
Subject: [PATCH] fix: improve error message for truncated file download

---
 powersimdata/output/output_data.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/powersimdata/output/output_data.py b/powersimdata/output/output_data.py
index 6cebf68be..54b477355 100644
--- a/powersimdata/output/output_data.py
+++ b/powersimdata/output/output_data.py
@@ -1,4 +1,5 @@
 import os
+import pickle
 
 import numpy as np
 import pandas as pd
@@ -41,6 +42,9 @@ def get_data(self, scenario_id, field_name):
         try:
             data = _read_data(file_name)
             return data
+        except pickle.UnpicklingError:
+            err_msg = f"Unable to unpickle {file_name}, possibly corrupted in download."
+            raise ValueError(err_msg)
         except FileNotFoundError:
             print(
                 "%s not found in %s on local machine"