Take dimension from TB model in tb.System

Author: greschd

z2pack/tb.py

@@ -21,7 +21,7 @@ class System(_HmSystem):
 
     :param kwargs:  Keyword arguments passed to :class:`.hm.System`.
 
-    The ``pos`` and ``bands`` keywords of :class:`.hm.System` are determined from the ``tb_model`` unless otherwise specified.
+    The ``pos``, ``bands`` and ``dim`` keywords of :class:`.hm.System` are determined from the ``tb_model`` unless otherwise specified.
     """
 
     def __init__(self, tb_model, **kwargs):
@@ -30,6 +30,10 @@ def __init__(self, tb_model, **kwargs):
             convention=2,
             **ChainMap(
                 kwargs,
-                dict(pos=copy.deepcopy(tb_model.pos), bands=tb_model.occ)
+                dict(
+                    pos=copy.deepcopy(tb_model.pos),
+                    bands=tb_model.occ,
+                    dim=tb_model.dim
+                )
             )
         )