blacken

Author: dhensle

activitysim/abm/models/trip_matrices.py

@@ -210,9 +210,7 @@ def write_trip_matrices(
         trips_df["origin"] = trips_df["true_origin"]
         trips_df["destination"] = trips_df["true_destination"]
         del trips_df["true_origin"], trips_df["true_destination"]
-        if (
-            network_los.zone_system == los.TWO_ZONE
-        ):
+        if network_los.zone_system == los.TWO_ZONE:
             trips_df["otaz"] = (
                 state.get_table("land_use").reindex(trips_df["origin"]).TAZ.tolist()
             )

activitysim/core/los.py

@@ -697,7 +697,7 @@ def get_tazs(self, state):
 
     def get_mazs(self):
         # FIXME - should compute on init?
-        assert self.zone_system  == TWO_ZONE
+        assert self.zone_system == TWO_ZONE
         mazs = self.maz_taz_df.MAZ.values
         assert isinstance(mazs, np.ndarray)
         return mazs