Browse Source

Fix flake8 warnings

pull/1520/head
Hironsan 3 years ago
parent
commit
1977ff165f
3 changed files with 8 additions and 6 deletions
  1. 2
      backend/api/models.py
  2. 4
      backend/api/tasks.py
  3. 8
      backend/api/views/label.py

2
backend/api/models.py

@ -1,8 +1,6 @@
import string
from typing import Literal
from django import db
from auto_labeling_pipeline.models import RequestModelFactory
from django.contrib.auth.models import User
from django.core.exceptions import ValidationError

4
backend/api/tasks.py

@ -1,6 +1,6 @@
import itertools
from celery import shared_task
from celery import shared_task
from celery.utils.log import get_task_logger
from django.conf import settings
from django.contrib.auth import get_user_model
@ -15,6 +15,8 @@ from .views.upload.factory import (get_data_class, get_dataset_class,
from .views.upload.utils import append_field
logger = get_task_logger(__name__)
class Buffer:
def __init__(self, buffer_size=settings.IMPORT_BATCH_SIZE):

8
backend/api/views/label.py

@ -15,12 +15,14 @@ from ..models import Label, Project
from ..permissions import IsInProjectReadOnlyOrAdmin, IsProjectAdmin
from ..serializers import LabelSerializer
def camel_to_snake(name):
name = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
return re.sub('([a-z0-9])([A-Z])', r'\1_\2', name).lower()
name = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
return re.sub('([a-z0-9])([A-Z])', r'\1_\2', name).lower()
def camel_to_snake_dict(d):
return {camel_to_snake(k): v for k,v in d.items()}
return {camel_to_snake(k): v for k, v in d.items()}
class LabelList(generics.ListCreateAPIView):

Loading…
Cancel
Save