improve formatting pandas-dev#28247

Author: sidharthann

pandas/core/groupby/generic.py

@@ -1904,9 +1904,8 @@ def _recast_datetimelike_result(result: DataFrame) -> DataFrame:
     """
     result = result.copy()
 
-    obj_cols = [
-        idx for idx in range(len(result.columns)) if is_object_dtype(result.dtypes.iloc[idx])
-    ]
+    obj_cols = [idx for idx in range(len(result.columns))
+                if is_object_dtype(result.dtypes.iloc[idx])]
 
     # See GH#26285
     for n in obj_cols:

pandas/tests/groupby/test_apply.py

@@ -665,5 +665,6 @@ def test_apply_datetime_issue():
     df = pd.DataFrame({'a': ['foo'], 'b': [datetime.today()]})
     result = df.groupby('a').apply(lambda x: pd.Series(['spam'], index=[42]))
 
-    expected = pd.DataFrame(['spam'], Index(['foo'], dtype='object', name='a'), columns=[42])
+    expected = pd.DataFrame(['spam'], Index(['foo'], dtype='object', name='a'),
+                            columns=[42])
     tm.assert_frame_equal(result, expected)