Update pypi workflow

.github/workflows/pypi.yml

@@ -7,6 +7,18 @@ on:
   workflow_dispatch:
 
 jobs:
+  build:
+    name: Build wheel distribution
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: 3.7
+    - name: Build a binary wheel and a source tarball
+      run: python setup.py sdist bdist_wheel
+
   pypi-publish:
     name: upload to pypi
     runs-on: ubuntu-latest
@@ -16,6 +28,5 @@ jobs:
       # IMPORTANT: this permission is mandatory for trusted publishing
       id-token: write
     steps:
-      # retrieve your distributions here
       - name: Publish package distributions to pypi
         uses: pypa/gh-action-pypi-publish@release/v1

fuxictr/pytorch/models/rank_model.py

@@ -177,7 +177,7 @@ def checkpoint_and_earlystop(self, logs, min_delta=1e-6):
                 self.save_weights(self.checkpoint)
         if self._stopping_steps >= self._early_stop_patience:
             self._stop_training = True
-            logging.info("********* Epoch=={} early stop *********".format(self._epoch_index + 1))
+            logging.info("********* Epoch={} early stop *********".format(self._epoch_index + 1))
         if not self._save_best_only:
             self.save_weights(self.checkpoint)