Merge pull request #181 from dstansby/fix-major

Fix default major
MAVENSDC · May 22, 2023 · deedfbc · deedfbc
2 parents 49c226d + d4d17ad
commit deedfbc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cdflib/cdfwrite.py b/cdflib/cdfwrite.py
@@ -181,7 +181,7 @@ def __init__(self, path, cdf_spec=None, delete=False):
 
         major = 1
         if cdf_spec is not None:
-            major = cdf_spec.get('Majority', 'column_major')
+            major = cdf_spec.get('Majority', major)
             if (isinstance(major, str)):
                 major = self._majority_token(major)
 
@@ -208,8 +208,8 @@ def __init__(self, path, cdf_spec=None, delete=False):
             num_rdim = 0
             rdim_sizes = None
 
-        if (major < 1 or major > 2):
-            raise OSError('Bad major.')
+        if major not in [1, 2]:
+            raise RuntimeError(f'Bad major: {major}')
 
         osSystem = pf.system()
         osMachine = pf.uname()[5]