Fix ordering of max_chi calc in change_chi method

Author: JanLuca

varipeps/peps/tensor.py

@@ -452,13 +452,13 @@ def change_chi(
           :obj:`~varipeps.peps.PEPS_Tensor`:
             New instance of the class with the increased value.
         """
-        if new_chi > self.max_chi:
+        new_max_chi = new_chi if reset_max_chi else self.max_chi
+
+        if new_chi > new_max_chi:
             raise ValueError(
                 "Increase above the max value for environment bond dimension."
             )
 
-        new_max_chi = new_chi if reset_max_chi else self.max_chi
-
         if new_chi < self.chi and reinitialize_env_as_identities:
             return type(self)(
                 tensor=self.tensor,
@@ -1407,13 +1407,13 @@ def change_chi(
           :obj:`~varipeps.peps.PEPS_Tensor`:
             New instance of the class with the increased value.
         """
-        if new_chi > self.max_chi:
+        new_max_chi = new_chi if reset_max_chi else self.max_chi
+
+        if new_chi > new_max_chi:
             raise ValueError(
                 "Increase above the max value for environment bond dimension."
             )
 
-        new_max_chi = new_chi if reset_max_chi else self.max_chi
-
         if new_chi < self.chi and reinitialize_env_as_identities:
             return type(self)(
                 tensor=self.tensor,
@@ -2330,13 +2330,13 @@ def change_chi(
           :obj:`~peps_ad.peps.PEPS_Tensor_Split_Transfer`:
             New instance of the class with the increased value.
         """
-        if new_chi > self.max_chi:
+        new_max_chi = new_chi if reset_max_chi else self.max_chi
+
+        if new_chi > new_max_chi:
             raise ValueError(
                 "Increase above the max value for environment bond dimension."
             )
 
-        new_max_chi = new_chi if reset_max_chi else self.max_chi
-
         new_interlayer_chi = new_chi if reset_interlayer_chi else self.interlayer_chi
 
         if new_chi < self.chi and reinitialize_env_as_identities: