Skip to content

Commit 02dbefb

Browse files
committed
Changed naming scheme; Renamed utils
1 parent 05d248d commit 02dbefb

File tree

3 files changed

+4
-9
lines changed

3 files changed

+4
-9
lines changed

flamingo_tools/extract_block.py renamed to flamingo_tools/extract_block_util.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -50,13 +50,8 @@ def extract_block(
5050

5151
if s3:
5252
image_name = input_content[-1].split(".")[0]
53-
if len(image_name.split("_")) > 1:
54-
resized_suffix = "_resized"
55-
image_prefix = image_name.split("_")[0]
56-
else:
57-
resized_suffix = ""
58-
image_prefix = image_name
59-
basename = input_content[0] + resized_suffix
53+
image_prefix = image_name
54+
basename = input_content[0]
6055
else:
6156
basename = "".join(input_content[-1].split(".")[:-1])
6257
image_prefix = basename.split("_")[-1]

reproducibility/block_extraction/repro_block_extraction.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import os
44
from typing import Optional
55

6-
from flamingo_tools.extract_block import extract_block
6+
from flamingo_tools.extract_block_util import extract_block
77

88

99
def repro_block_extraction(

scripts/extract_block.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import argparse
44
import json
55

6-
from flamingo_tools.extract_block import extract_block
6+
from flamingo_tools.extract_block_util import extract_block
77

88

99
def main():

0 commit comments

Comments
 (0)