Better way to avoid flake8 undefined name error

python_scripts/linear_models_ex_03.py

@@ -79,10 +79,10 @@
 # Write your code here.
 
 # %% [markdown]
-# For the following questions, you can use the following snippet to get the
-# feature names from the column transformer here named `preprocessor`.
-
-# %%
+# For the following questions, you can copy adn paste the following snippet to
+# get the feature names from the column transformer here named `preprocessor`.
+#
+# ```python
 # preprocessor.fit(data)
 # feature_names = (
 #     preprocessor.named_transformers_["onehotencoder"].get_feature_names_out(
@@ -91,6 +91,10 @@
 # ).tolist()
 # feature_names += numerical_columns
 # feature_names
+# ```
+
+# %%
+# Write your code here.
 
 # %% [markdown]
 # Notice that there are as many feature names as coefficients in the last step

python_scripts/linear_models_sol_03.py

@@ -145,10 +145,22 @@
 )
 
 # %% [markdown]
-# For the following questions, you can use the following snippet to get the
-# feature names from the column transformer here named `preprocessor`.
+# For the following questions, you can copy adn paste the following snippet to
+# get the feature names from the column transformer here named `preprocessor`.
+#
+# ```python
+# preprocessor.fit(data)
+# feature_names = (
+#     preprocessor.named_transformers_["onehotencoder"].get_feature_names_out(
+#         categorical_columns
+#     )
+# ).tolist()
+# feature_names += numerical_columns
+# feature_names
+# ```
 
 # %%
+# solution
 preprocessor.fit(data)
 feature_names = (
     preprocessor.named_transformers_["onehotencoder"].get_feature_names_out(