From e46b82195af1ae209bfee5fc41dfb3723a700b1b Mon Sep 17 00:00:00 2001 From: Daniel Olsen Date: Mon, 21 Dec 2020 10:56:47 -0800 Subject: [PATCH] refactor: use ternary if/else assignment for n_storage --- powersimdata/input/scenario_grid.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/powersimdata/input/scenario_grid.py b/powersimdata/input/scenario_grid.py index f512f7e92..ed842aab4 100644 --- a/powersimdata/input/scenario_grid.py +++ b/powersimdata/input/scenario_grid.py @@ -42,10 +42,7 @@ def _read_network(self): try: # The next line will fail if no iess attribute (index energy storage system) # Since we use the 'squeeze_me' param, 1 storage -> an int, not an array - if isinstance(mpc.iess, int): - n_storage = 1 - else: - n_storage = mpc.iess.shape[0] + n_storage = 1 if isinstance(mpc.iess, int) else mpc.iess.shape[0] except AttributeError: n_storage = 0