diff --git a/samples/model-builder/create_entity_type_sample.py b/samples/model-builder/create_entity_type_sample.py
index 2a1c4da7fd..a1cfc0e6d3 100644
--- a/samples/model-builder/create_entity_type_sample.py
+++ b/samples/model-builder/create_entity_type_sample.py
@@ -18,10 +18,7 @@
 
 
 def create_entity_type_sample(
-    project: str,
-    location: str,
-    entity_type_id: str,
-    featurestore_name: str,
+    project: str, location: str, entity_type_id: str, featurestore_name: str,
 ):
 
     aiplatform.init(project=project, location=location)
@@ -33,4 +30,6 @@ def create_entity_type_sample(
     fs.wait()
 
     return fs
+
+
 #  [END aiplatform_sdk_create_entity_type_sample]