Allow black to reformat correctly

moto/ecs/models.py

@@ -134,7 +134,7 @@ class TaskDefinition(BaseObject):
         else:
             self.volumes = volumes
         if network_mode is None:
-            self.network_mode = 'bridge'
+            self.network_mode = "bridge"
         else:
             self.network_mode = network_mode
 
@@ -567,7 +567,13 @@ class EC2ContainerServiceBackend(BaseBackend):
             self.task_definitions[family] = {}
             revision = 1
         task_definition = TaskDefinition(
-            family, revision, container_definitions, self.region_name, volumes=volumes, network_mode=network_mode, tags=tags
+            family,
+            revision,
+            container_definitions,
+            self.region_name,
+            volumes=volumes,
+            network_mode=network_mode,
+            tags=tags,
         )
         self.task_definitions[family][revision] = task_definition
 

moto/ecs/responses.py

@@ -62,9 +62,13 @@ class EC2ContainerServiceResponse(BaseResponse):
         container_definitions = self._get_param("containerDefinitions")
         volumes = self._get_param("volumes")
         tags = self._get_param("tags")
-        network_mode = self._get_param('networkMode')
+        network_mode = self._get_param("networkMode")
         task_definition = self.ecs_backend.register_task_definition(
-            family, container_definitions, volumes=volumes, network_mode=network_mode, tags=tags,
+            family,
+            container_definitions,
+            volumes=volumes,
+            network_mode=network_mode,
+            tags=tags,
         )
         return json.dumps({"taskDefinition": task_definition.response_object})
 

tests/test_ecs/test_ecs_boto3.py

@@ -94,7 +94,7 @@ def test_register_task_definition():
                 "logConfiguration": {"logDriver": "json-file"},
             }
         ],
-        networkMode='bridge',
+        networkMode="bridge",
         tags=[
             {"key": "createdBy", "value": "moto-unittest"},
             {"key": "foo", "value": "bar"},
@@ -125,7 +125,7 @@ def test_register_task_definition():
     response["taskDefinition"]["containerDefinitions"][0]["logConfiguration"][
         "logDriver"
     ].should.equal("json-file")
-    response['taskDefinition']['networkMode'].should.equal('bridge')
+    response["taskDefinition"]["networkMode"].should.equal("bridge")
 
 
 @mock_ecs