Update author information in setup.py and enhance forward method sign…

…ature in Loss class for improved readability

balanced_loss/losses.py

@@ -44,7 +44,7 @@ def __init__(
         fl_gamma=2,
         samples_per_class=None,
         class_balanced=False,
-        safe: bool = False
+        safe: bool = False,
     ):
         """
         Compute the Class Balanced Loss between `logits` and the ground truth `labels`.
@@ -77,11 +77,7 @@ def __init__(
         self.class_balanced = class_balanced
         self.safe = safe
 
-    def forward(
-        self,
-        logits: torch.tensor,
-        labels: torch.tensor
-    ):
+    def forward(self, logits: torch.tensor, labels: torch.tensor):
         """
         Compute the Class Balanced Loss between `logits` and the ground truth `labels`.
         Class Balanced Loss: ((1-beta)/(1-beta^n))*Loss(labels, logits)

setup.py

@@ -37,7 +37,7 @@ def get_version():
 setuptools.setup(
     name="balanced-loss",
     version=get_version(),
-    author="",
+    author="fcakyon",
     license="MIT",
     description="Easy to use class-balanced cross-entropy and focal loss implementation for Pytorch.",
     long_description=get_long_description(),
@@ -54,9 +54,9 @@ def get_version():
         "Intended Audience :: Developers",
         "Intended Audience :: Science/Research",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
         "Topic :: Software Development :: Libraries",
         "Topic :: Software Development :: Libraries :: Python Modules",
         "Topic :: Education",