Skip to content

Commit

Permalink
fixes #78: Update model module documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
Maxence Guindon committed Apr 24, 2024
1 parent 7664fca commit 569d624
Show file tree
Hide file tree
Showing 5 changed files with 13 additions and 12 deletions.
4 changes: 4 additions & 0 deletions model/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
"""
This module provides functions for requesting inferences from different models.
"""

from model.swin import request_inference_from_swin
from model.seed_detector import request_inference_from_seed_detector
from model.test import request_inference_from_test
Expand Down
4 changes: 2 additions & 2 deletions model/seed_detector.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""
This file contains the function that requests the inference and processes the data from
the seed detector model.
This file contains the function that requests the inference and processes
the data from the seed detector model.
"""

import io
Expand Down
5 changes: 3 additions & 2 deletions model/six_seeds.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
"""
This file contains the function that requests the inference and processes the data from
the nachet-6seeds model.
This file contains the function that requests the inference and processes
the data from the nachet-6seeds model.
"""

import json

from collections import namedtuple
from urllib.request import Request, urlopen, HTTPError
from custom_exceptions import ProcessInferenceResultError
Expand Down
4 changes: 2 additions & 2 deletions model/swin.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""
This file contains the function that requests the inference and processes the data from
the swin model.
This file contains the function that requests the inference and processes
the data from the swin model.
"""

import json
Expand Down
8 changes: 2 additions & 6 deletions model/test.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,6 @@
"""
This module contains functions for performing inference using different models.
Functions:
request_inference_from_swin: Perform inference using the SWIN model on a list of images.
request_inference_from_seed_detector: Requests inference from the seed detector model using the provided previous result.
request_inference_from_nachet_six_seed: Requests inference from the Nachet Six Seed model.
This module contains functions for testing the inference procedure in
the backend.
"""
from collections import namedtuple
from custom_exceptions import ProcessInferenceResultError
Expand Down

0 comments on commit 569d624

Please sign in to comment.