autogluon.multimodal 1.1.2b20240820__tar.gz → 1.1.2b20240821__tar.gz

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.multimodal
-Version: 1.1.2b20240820
+Version: 1.1.2b20240821
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/setup.py

@@ -60,8 +60,7 @@ install_requires = [
 install_requires = ag.get_dependency_version_ranges(install_requires)
 
 tests_require = [
-    "black~=23.0",
-    "isort>=5.10",
+    "ruff",
     "datasets>=2.10.0,<2.15.0",
     "onnx>=1.13.0,<1.14.0",
     "onnxruntime>=1.15.0,<1.16.0;platform_system=='Darwin'",

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/configs/pretrain/detection/dino/dino-4scale_r50_8xb2-12e_coco.py

@@ -49,7 +49,10 @@ model = dict(
         post_norm_cfg=None,
     ),
     positional_encoding=dict(
-        num_feats=128, normalize=True, offset=0.0, temperature=20  # -0.5 for DeformDETR
+        num_feats=128,
+        normalize=True,
+        offset=0.0,
+        temperature=20,  # -0.5 for DeformDETR
     ),  # 10000 for DeformDETR
     bbox_head=dict(
         type="DINOHead",

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/configs/pretrain/detection/dino/dino-5scale_swin-l_8xb2-12e_coco.py

@@ -1,6 +1,8 @@
 _base_ = "./dino-4scale_r50_8xb2-12e_coco.py"
 
-pretrained = "https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth"  # noqa
+pretrained = (
+    "https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth"  # noqa
+)
 num_levels = 5
 model = dict(
     num_feature_levels=num_levels,

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/configs/pretrain/detection/faster_rcnn/faster_rcnn_r50_fpn.py

@@ -79,7 +79,7 @@ model = dict(
     ),
     test_cfg=dict(
         rpn=dict(nms_pre=1000, max_per_img=1000, nms=dict(type="nms", iou_threshold=0.7), min_bbox_size=0),
-        rcnn=dict(score_thr=0.05, nms=dict(type="nms", iou_threshold=0.5), max_per_img=100)
+        rcnn=dict(score_thr=0.05, nms=dict(type="nms", iou_threshold=0.5), max_per_img=100),
         # soft-nms is also supported for rcnn testing
         # e.g., nms=dict(type='soft_nms', iou_threshold=0.5, min_score=0.05)
     ),

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/constants.py

@@ -1,4 +1,5 @@
 """Storing the constants"""
+
 from autogluon.core.metrics import METRICS
 
 # Column/Label Types

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/data/infer_types.py

@@ -828,9 +828,7 @@ def infer_ner_column_type(column_types: Dict):
     if any([col_type.startswith(TEXT_NER) for col_type in column_types.values()]):
         return column_types
 
-    for (
-        column
-    ) in (
+    for column in (
         column_types.keys()
     ):  # column_types is an ordered dict, so column_types.keys() returns the keys in the order of insertions.
         if column_types[column].startswith(TEXT):

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/data/process_semantic_seg_img.py

@@ -235,7 +235,9 @@ class SemanticSegImageProcessor:
                 if self.num_classes == 1:
                     gt = gt.convert("L")
                 if self.num_classes > 1:
-                    gt = np.array(gt).astype(
+                    gt = np.array(
+                        gt
+                    ).astype(
                         "float32"
                     )  # There may be issues with 'transforms.ToTensor()' without this line because 'transforms.ToTensor()' converts 'unit8' to values between 0 and 1.
                     gt = Image.fromarray(gt)

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/data/trivial_augmenter.py

@@ -1,5 +1,5 @@
 """
-This file implements TrivialAugment.(https://arxiv.org/abs/2103.10158) We extend it for multi-modality setting. 
+This file implements TrivialAugment.(https://arxiv.org/abs/2103.10158) We extend it for multi-modality setting.
 
 Code is partially adapted from its official implementation https://github.com/automl/trivialaugment
 """

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/learners/base.py

@@ -1017,8 +1017,8 @@ class BaseLearner(ExportMixin, DistillationMixin, RealtimeMixin):
         if (
             config.env.strategy == DEEPSPEED_OFFLOADING and num_gpus == 1 and DEEPSPEED_MODULE not in sys.modules
         ):  # Offloading currently only tested for single GPU
-            assert version.parse(pl.__version__) >= version.parse(
-                DEEPSPEED_MIN_PL_VERSION
+            assert (
+                version.parse(pl.__version__) >= version.parse(DEEPSPEED_MIN_PL_VERSION)
             ), f"For DeepSpeed Offloading to work reliably you need at least lightning version {DEEPSPEED_MIN_PL_VERSION}, however, found {pl.__version__}. Please update your lightning version."
             from ..optimization.deepspeed import CustomDeepSpeedStrategy
 

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/models/custom_hf_models/modeling_sam_for_conv_lora.py

@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch SAM model."""
+"""PyTorch SAM model."""
 
 import collections
 import math

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/optimization/lit_distiller.py

@@ -416,7 +416,7 @@ class DistillerLitModule(pl.LightningModule):
             custom_metric_func=self.custom_metric_func,
             logits=student_output[self.student_model.prefix][LOGITS],
             label=batch[self.student_model.label_key],
-        ),
+        )
         self.log(
             self.validation_metric_name,
             self.validation_metric,

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/optimization/lit_module.py

@@ -296,7 +296,7 @@ class LitModule(pl.LightningModule):
             custom_metric_func=self.custom_metric_func,
             logits=output[self.model.prefix][LOGITS],
             label=batch[self.model.label_key],
-        ),
+        )
         self.log(
             self.validation_metric_name,
             self.validation_metric,

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/optimization/lit_ner.py

@@ -189,7 +189,7 @@ class NerLitModule(LitModule):
             custom_metric_func=self.custom_metric_func,
             logits=active_logits,
             label=active_labels,
-        ),
+        )
         self.log(
             self.validation_metric_name,
             self.validation_metric,

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/optimization/lr_scheduler.py

@@ -2,6 +2,7 @@
 This file is largely borrowed from `transformers/optimization.py`.
 This is to make lr schedulers pickle-able so that we can use the training strategy "ddp_spawn" in Pytorch Lightning.
 """
+
 import functools
 import math
 

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/optimization/utils.py

@@ -852,8 +852,8 @@ def gather_column_features(
 
             # two or more columns share one cls feature, and no other columns share it.
             if len(columns_share_one_feature) > 0:
-                assert len("_".join(columns_share_one_feature)) == len(
-                    feature_name
+                assert (
+                    len("_".join(columns_share_one_feature)) == len(feature_name)
                 ), f"model `{per_model_name}`'s cls feature name `{feature_name}` doesn't match `{columns_share_one_feature}`"
                 gathered_features.append(per_model_output[COLUMN_FEATURES][FEATURES][feature_name])
 

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/problem_types.py

@@ -250,8 +250,8 @@ _ner_property = ProblemTypeProperty(
     _supported_evaluation_metrics=[OVERALL_F1, NER_TOKEN_F1],
     _fallback_validation_metric=NER_TOKEN_F1,
 )
-PROBLEM_TYPES_REG.register(NER, _ner_property),
-PROBLEM_TYPES_REG.register(NAMED_ENTITY_RECOGNITION, _ner_property),
+PROBLEM_TYPES_REG.register(NER, _ner_property)
+PROBLEM_TYPES_REG.register(NAMED_ENTITY_RECOGNITION, _ner_property)
 
 # Feature Extraction: text --> feature, image --> features
 PROBLEM_TYPES_REG.register(

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/registry.py

@@ -110,6 +110,7 @@ class Registry:
             return self.create(name, **args)
         else:
             raise NotImplementedError(
-                "The format of json string is not supported! We only support "
-                'list/dict. json_str="{}".'.format(json_str)
+                "The format of json string is not supported! We only support " 'list/dict. json_str="{}".'.format(
+                    json_str
+                )
             )

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/checkpoint.py

@@ -9,9 +9,8 @@ import torch
 from lightning.pytorch.strategies import DeepSpeedStrategy
 from lightning.pytorch.utilities.rank_zero import rank_zero_warn
 
-from .cloud_io import _atomic_save
+from .cloud_io import _atomic_save, get_filesystem
 from .cloud_io import _load as pl_load
-from .cloud_io import get_filesystem
 
 logger = logging.getLogger(__name__)
 

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/cloud_io.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Utilities related to data saving/loading."""
+
 """
 Copied from
 https://github.com/Lightning-AI/lightning/blob/master/src/lightning/fabric/utilities/cloud_io.py

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/colormap.py

@@ -6,6 +6,7 @@
 An awesome colormap for really neat visualizations.
 Copied from Detectron, and removed gray colors.
 """
+
 import random
 
 import numpy as np

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/download.py

@@ -98,9 +98,7 @@ def download(
                 finally:
                     raise OSError(
                         "Moving downloaded temp file - {}, to {} failed. \
-                        Please retry the download.".format(
-                            src, dst
-                        )
+                        Please retry the download.".format(src, dst)
                     )
 
     else:

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/hpo.py

@@ -50,9 +50,9 @@ def hpo_trial(sampled_hyperparameters, learner, checkpoint_dir=None, **_fit_args
     resources = context.get_trial_resources().required_resources
     num_cpus = int(resources.get("CPU"))
 
-    _fit_args[
-        "hyperparameters"
-    ] = sampled_hyperparameters  # The original hyperparameters is the search space, replace it with the hyperparameters sampled
+    _fit_args["hyperparameters"] = (
+        sampled_hyperparameters  # The original hyperparameters is the search space, replace it with the hyperparameters sampled
+    )
     _fit_args["save_path"] = context.get_trial_dir()  # We want to save each trial to a separate directory
     logger.debug(f"hpo trial save_path: {_fit_args['save_path']}")
     if checkpoint_dir is not None:

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/nlpaug.py

@@ -1,6 +1,7 @@
 """Utilities built on top of nlpaug. This file is meant to be lazy-imported because importing nlpaug can take time.
 See the discussion in https://github.com/autogluon/autogluon/issues/2706
 """
+
 import random
 
 from nlpaug import Augmenter

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/utils/visualizer.py

@@ -578,9 +578,7 @@ class NERVisualizer:
         """
         text = '<mark style="background-color:{}; color:white; border-radius: .6em .6em; padding: .1em;">{} \
          <b style="background-color:white; color:black; font-size:x-small; border-radius: 0.5em .5em; padding: .0em;">{} </b> \
-         </mark>'.format(
-            color, self.escape_html(text), self.escape_html(label)
-        )
+         </mark>'.format(color, self.escape_html(text), self.escape_html(label))
         return text
 
     def _repr_html_(self):

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon/multimodal/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.1.2b20240820'
+__version__ = '1.1.2b20240821'
 __lite__ = False

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon.multimodal.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.multimodal
-Version: 1.1.2b20240820
+Version: 1.1.2b20240821
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.multimodal-1.1.2b20240820 → autogluon.multimodal-1.1.2b20240821}/src/autogluon.multimodal.egg-info/requires.txt

@@ -20,9 +20,9 @@ text-unidecode<1.4,>=1.3
 torchmetrics<1.3.0,>=1.2.0
 nptyping<2.5.0,>=1.4.4
 omegaconf<2.3.0,>=2.1.1
-autogluon.core[raytune]==1.1.2b20240820
-autogluon.features==1.1.2b20240820
-autogluon.common==1.1.2b20240820
+autogluon.core[raytune]==1.1.2b20240821
+autogluon.features==1.1.2b20240821
+autogluon.common==1.1.2b20240821
 pytorch-metric-learning<2.4,>=1.3.0
 nlpaug<1.2.0,>=1.1.10
 nltk<4.0.0,>=3.4.5
@@ -35,8 +35,7 @@ nvidia-ml-py3==7.352.0
 pdf2image<1.19,>=1.17.0
 
 [tests]
-black~=23.0
-isort>=5.10
+ruff
 datasets<2.15.0,>=2.10.0
 onnx<1.14.0,>=1.13.0