From 24268c2021ad565cc69ee4f434c63083d4755416 Mon Sep 17 00:00:00 2001
From: Kevin Hu <6051736+kevinhu@users.noreply.github.com>
Date: Fri, 11 Jun 2021 09:44:18 -0700
Subject: [PATCH] feat(ingest): headers for codegen Python scripts (#2637)

---
 metadata-ingestion/scripts/avro_codegen.py    | 28 +++++++++++++++----
 .../src/datahub/metadata/__init__.py          |  7 +++++
 .../src/datahub/metadata/com/__init__.py      |  7 +++++
 .../datahub/metadata/com/linkedin/__init__.py |  7 +++++
 .../metadata/com/linkedin/events/__init__.py  |  7 +++++
 .../com/linkedin/pegasus2avro/__init__.py     |  7 +++++
 .../linkedin/pegasus2avro/chart/__init__.py   |  7 +++++
 .../linkedin/pegasus2avro/common/__init__.py  |  7 +++++
 .../common/fieldtransformer/__init__.py       |  7 +++++
 .../pegasus2avro/dashboard/__init__.py        |  7 +++++
 .../linkedin/pegasus2avro/datajob/__init__.py |  7 +++++
 .../pegasus2avro/datajob/azkaban/__init__.py  |  7 +++++
 .../pegasus2avro/dataplatform/__init__.py     |  7 +++++
 .../pegasus2avro/dataprocess/__init__.py      |  7 +++++
 .../linkedin/pegasus2avro/dataset/__init__.py |  7 +++++
 .../pegasus2avro/glossary/__init__.py         |  7 +++++
 .../pegasus2avro/identity/__init__.py         |  7 +++++
 .../pegasus2avro/metadata/__init__.py         |  7 +++++
 .../pegasus2avro/metadata/key/__init__.py     |  7 +++++
 .../metadata/snapshot/__init__.py             |  7 +++++
 .../com/linkedin/pegasus2avro/ml/__init__.py  |  7 +++++
 .../pegasus2avro/ml/metadata/__init__.py      |  7 +++++
 .../com/linkedin/pegasus2avro/mxe/__init__.py |  7 +++++
 .../linkedin/pegasus2avro/schema/__init__.py  |  7 +++++
 .../com/linkedin/pegasus2avro/tag/__init__.py |  7 +++++
 .../src/datahub/metadata/schema_classes.py    |  4 +++
 26 files changed, 194 insertions(+), 6 deletions(-)

diff --git a/metadata-ingestion/scripts/avro_codegen.py b/metadata-ingestion/scripts/avro_codegen.py
index e894b05c5a..d25cc0cd69 100644
--- a/metadata-ingestion/scripts/avro_codegen.py
+++ b/metadata-ingestion/scripts/avro_codegen.py
@@ -1,18 +1,31 @@
 import json
+from pathlib import Path
+from typing import Union
 
 import click
 from avrogen import write_schema_files
 
+autogen_header = """# flake8: noqa
 
-def suppress_checks_in_file(filepath: str) -> None:
-    """Adds a couple lines to the top of a file to suppress flake8 and black"""
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+"""
+
+
+def suppress_checks_in_file(filepath: Union[str, Path]) -> None:
+    """
+    Adds a couple lines to the top of an autogenerated file:
+        - Comments to suppress flake8 and black.
+        - A note stating that the file was autogenerated.
+    """
 
     with open(filepath, "r+") as f:
         contents = f.read()
 
         f.seek(0, 0)
-        f.write("# flake8: noqa\n")
-        f.write("# fmt: off\n")
+        f.write(autogen_header)
         f.write(contents)
         f.write("# fmt: on\n")
 
@@ -21,7 +34,6 @@ def suppress_checks_in_file(filepath: str) -> None:
 @click.argument("schema_file", type=click.Path(exists=True))
 @click.argument("outdir", type=click.Path())
 def generate(schema_file: str, outdir: str) -> None:
-    # print(f'using {schema_file}')
     with open(schema_file) as f:
         raw_schema_text = f.read()
 
@@ -33,11 +45,15 @@ def generate(schema_file: str, outdir: str) -> None:
     redo_spaces = json.dumps(json.loads(schema_json), indent=2)
 
     write_schema_files(redo_spaces, outdir)
-    suppress_checks_in_file(f"{outdir}/schema_classes.py")
     with open(f"{outdir}/__init__.py", "w"):
         # Truncate this file.
         pass
 
+    # Add headers for all generated files
+    generated_files = Path(outdir).glob("**/*.py")
+    for file in generated_files:
+        suppress_checks_in_file(file)
+
 
 if __name__ == "__main__":
     generate()
diff --git a/metadata-ingestion/src/datahub/metadata/__init__.py b/metadata-ingestion/src/datahub/metadata/__init__.py
index e69de29bb2..144057cdbf 100644
--- a/metadata-ingestion/src/datahub/metadata/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/__init__.py
@@ -0,0 +1,7 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/__init__.py b/metadata-ingestion/src/datahub/metadata/com/__init__.py
index e69de29bb2..144057cdbf 100644
--- a/metadata-ingestion/src/datahub/metadata/com/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/__init__.py
@@ -0,0 +1,7 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/__init__.py
index e69de29bb2..144057cdbf 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/__init__.py
@@ -0,0 +1,7 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/events/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/events/__init__.py
index 2befa2ae6e..6bb82c355b 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/events/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/events/__init__.py
@@ -1,4 +1,11 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from ....schema_classes import KafkaAuditHeaderClass
 
 
 KafkaAuditHeader = KafkaAuditHeaderClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/__init__.py
index e69de29bb2..144057cdbf 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/__init__.py
@@ -0,0 +1,7 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/chart/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/chart/__init__.py
index 3e537f7ffd..cf0c6b430c 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/chart/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/chart/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import ChartInfoClass
 from .....schema_classes import ChartQueryClass
 from .....schema_classes import ChartQueryTypeClass
@@ -8,3 +14,4 @@ ChartInfo = ChartInfoClass
 ChartQuery = ChartQueryClass
 ChartQueryType = ChartQueryTypeClass
 ChartType = ChartTypeClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/__init__.py
index b359f31feb..ba4a250b7e 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import AccessLevelClass
 from .....schema_classes import AuditStampClass
 from .....schema_classes import BrowsePathsClass
@@ -48,3 +54,4 @@ OwnershipType = OwnershipTypeClass
 Status = StatusClass
 TagAssociation = TagAssociationClass
 VersionTag = VersionTagClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/fieldtransformer/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/fieldtransformer/__init__.py
index d7115bb723..4f1e2f797a 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/fieldtransformer/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/common/fieldtransformer/__init__.py
@@ -1,6 +1,13 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from ......schema_classes import TransformationTypeClass
 from ......schema_classes import UDFTransformerClass
 
 
 TransformationType = TransformationTypeClass
 UDFTransformer = UDFTransformerClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dashboard/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dashboard/__init__.py
index c7a50acf41..2924c13fcf 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dashboard/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dashboard/__init__.py
@@ -1,4 +1,11 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import DashboardInfoClass
 
 
 DashboardInfo = DashboardInfoClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/__init__.py
index 3d3f845745..c4cae2db0d 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import DataFlowInfoClass
 from .....schema_classes import DataJobInfoClass
 from .....schema_classes import DataJobInputOutputClass
@@ -6,3 +12,4 @@ from .....schema_classes import DataJobInputOutputClass
 DataFlowInfo = DataFlowInfoClass
 DataJobInfo = DataJobInfoClass
 DataJobInputOutput = DataJobInputOutputClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/azkaban/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/azkaban/__init__.py
index dbeccdd1d3..13c26a929b 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/azkaban/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/datajob/azkaban/__init__.py
@@ -1,4 +1,11 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from ......schema_classes import AzkabanJobTypeClass
 
 
 AzkabanJobType = AzkabanJobTypeClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataplatform/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataplatform/__init__.py
index 011c903e87..30b0a188a2 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataplatform/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataplatform/__init__.py
@@ -1,6 +1,13 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import DataPlatformInfoClass
 from .....schema_classes import PlatformTypeClass
 
 
 DataPlatformInfo = DataPlatformInfoClass
 PlatformType = PlatformTypeClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataprocess/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataprocess/__init__.py
index 4ff082f78c..e7abece20e 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataprocess/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataprocess/__init__.py
@@ -1,4 +1,11 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import DataProcessInfoClass
 
 
 DataProcessInfo = DataProcessInfoClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataset/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataset/__init__.py
index 2efda07731..0aa694f351 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataset/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/dataset/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import DatasetDeprecationClass
 from .....schema_classes import DatasetFieldMappingClass
 from .....schema_classes import DatasetLineageTypeClass
@@ -14,3 +20,4 @@ DatasetProperties = DatasetPropertiesClass
 DatasetUpstreamLineage = DatasetUpstreamLineageClass
 Upstream = UpstreamClass
 UpstreamLineage = UpstreamLineageClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/glossary/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/glossary/__init__.py
index 7455fd03bd..c75c1eda70 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/glossary/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/glossary/__init__.py
@@ -1,6 +1,13 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import GlossaryNodeInfoClass
 from .....schema_classes import GlossaryTermInfoClass
 
 
 GlossaryNodeInfo = GlossaryNodeInfoClass
 GlossaryTermInfo = GlossaryTermInfoClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/identity/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/identity/__init__.py
index 1621b73f4d..60fdfaa4e7 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/identity/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/identity/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import CorpGroupInfoClass
 from .....schema_classes import CorpUserEditableInfoClass
 from .....schema_classes import CorpUserInfoClass
@@ -6,3 +12,4 @@ from .....schema_classes import CorpUserInfoClass
 CorpGroupInfo = CorpGroupInfoClass
 CorpUserEditableInfo = CorpUserEditableInfoClass
 CorpUserInfo = CorpUserInfoClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/__init__.py
index e69de29bb2..144057cdbf 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/__init__.py
@@ -0,0 +1,7 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py
index 48353e186f..dac0136932 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/key/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from ......schema_classes import ChartKeyClass
 from ......schema_classes import CorpGroupKeyClass
 from ......schema_classes import CorpUserKeyClass
@@ -32,3 +38,4 @@ MLFeatureTableKey = MLFeatureTableKeyClass
 MLModelKey = MLModelKeyClass
 MLPrimaryKeyKey = MLPrimaryKeyKeyClass
 TagKey = TagKeyClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/snapshot/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/snapshot/__init__.py
index 9da58c6eba..90098c2765 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/snapshot/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/metadata/snapshot/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from ......schema_classes import ChartSnapshotClass
 from ......schema_classes import CorpGroupSnapshotClass
 from ......schema_classes import CorpUserSnapshotClass
@@ -32,3 +38,4 @@ MLFeatureTableSnapshot = MLFeatureTableSnapshotClass
 MLModelSnapshot = MLModelSnapshotClass
 MLPrimaryKeySnapshot = MLPrimaryKeySnapshotClass
 TagSnapshot = TagSnapshotClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/__init__.py
index e69de29bb2..144057cdbf 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/__init__.py
@@ -0,0 +1,7 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/metadata/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/metadata/__init__.py
index b8ffc51e21..7518ef2c7a 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/metadata/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/ml/metadata/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from ......schema_classes import BaseDataClass
 from ......schema_classes import CaveatDetailsClass
 from ......schema_classes import CaveatsAndRecommendationsClass
@@ -38,3 +44,4 @@ SourceCode = SourceCodeClass
 SourceCodeUrl = SourceCodeUrlClass
 SourceCodeUrlType = SourceCodeUrlTypeClass
 TrainingData = TrainingDataClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/mxe/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/mxe/__init__.py
index dfbeac91f8..10176a6def 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/mxe/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/mxe/__init__.py
@@ -1,4 +1,11 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import MetadataChangeEventClass
 
 
 MetadataChangeEvent = MetadataChangeEventClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/schema/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/schema/__init__.py
index 6be512dac1..46fdd935a7 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/schema/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/schema/__init__.py
@@ -1,3 +1,9 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import ArrayTypeClass
 from .....schema_classes import BinaryJsonSchemaClass
 from .....schema_classes import BooleanTypeClass
@@ -62,3 +68,4 @@ StringType = StringTypeClass
 TimeType = TimeTypeClass
 UnionType = UnionTypeClass
 UrnForeignKey = UrnForeignKeyClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/tag/__init__.py b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/tag/__init__.py
index 128503adc4..7f2005d47e 100644
--- a/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/tag/__init__.py
+++ b/metadata-ingestion/src/datahub/metadata/com/linkedin/pegasus2avro/tag/__init__.py
@@ -1,4 +1,11 @@
+# flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
+# fmt: off
 from .....schema_classes import TagPropertiesClass
 
 
 TagProperties = TagPropertiesClass
+# fmt: on
diff --git a/metadata-ingestion/src/datahub/metadata/schema_classes.py b/metadata-ingestion/src/datahub/metadata/schema_classes.py
index 5fea1d6d3e..54a65f1532 100644
--- a/metadata-ingestion/src/datahub/metadata/schema_classes.py
+++ b/metadata-ingestion/src/datahub/metadata/schema_classes.py
@@ -1,4 +1,8 @@
 # flake8: noqa
+
+# This file is autogenerated by /metadata-ingestion/scripts/avro_codegen.py
+# Do not modify manually!
+
 # fmt: off
 import json
 import os.path