diff --git a/.github/workflows/format.yml b/.github/workflows/format.yml
index 81e1018..dd6111f 100644
--- a/.github/workflows/format.yml
+++ b/.github/workflows/format.yml
@@ -6,9 +6,9 @@ name: Ultralytics Actions
 
 on:
   push:
-    branches: [main,master]
+    branches: [main]
   pull_request:
-    branches: [main,master]
+    branches: [main]
 
 jobs:
   format:
diff --git a/train_tf.py b/train_tf.py
index 9b80c23..d483a2c 100644
--- a/train_tf.py
+++ b/train_tf.py
@@ -4,7 +4,6 @@
 import plotly.graph_objs as go
 import scipy.io
 import tensorflow as tf
-import tensorflow.contrib.eager as tfe
 from plotly.offline import plot
 
 from utils.utils import *