smodusermc commited on
Commit
aefdaa1
·
verified ·
1 Parent(s): d7f1272

Update storage_handler.py

Browse files
Files changed (1) hide show
  1. storage_handler.py +4 -4
storage_handler.py CHANGED
@@ -76,8 +76,9 @@ _fs = HfFileSystem(token=HF_TOKEN)
76
  # ------------------------------------------------------------------------
77
  def ensure_bucket():
78
  """Create the private bucket if it doesn't exist."""
 
 
79
  try:
80
- # Try to create bucket
81
  _api.create_bucket(
82
  bucket_id=f"{OWNER}/{BUCKET_ID}",
83
  private=True,
@@ -92,15 +93,14 @@ def ensure_bucket():
92
  logger.warning(f"⚠️ Cannot create bucket - permission issue: {e}")
93
  logger.info("The bucket may already exist or you don't have permissions")
94
  else:
95
- # Try alternative bucket name format
96
  try:
97
  _api.create_bucket(
98
  bucket_id=BUCKET_ID,
99
  private=True,
100
  exist_ok=True
101
  )
102
- global OWNER, BUCKET_URI, BUCKET_ID
103
- OWNER = "" # No owner prefix
104
  BUCKET_URI = f"hf://buckets/{BUCKET_ID}"
105
  logger.info(f"✅ Bucket '{BUCKET_ID}' is ready (alternative format)")
106
  except Exception as e2:
 
76
  # ------------------------------------------------------------------------
77
  def ensure_bucket():
78
  """Create the private bucket if it doesn't exist."""
79
+ global OWNER, BUCKET_URI, BUCKET_ID
80
+
81
  try:
 
82
  _api.create_bucket(
83
  bucket_id=f"{OWNER}/{BUCKET_ID}",
84
  private=True,
 
93
  logger.warning(f"⚠️ Cannot create bucket - permission issue: {e}")
94
  logger.info("The bucket may already exist or you don't have permissions")
95
  else:
 
96
  try:
97
  _api.create_bucket(
98
  bucket_id=BUCKET_ID,
99
  private=True,
100
  exist_ok=True
101
  )
102
+ # now we can safely reassign globals
103
+ OWNER = ""
104
  BUCKET_URI = f"hf://buckets/{BUCKET_ID}"
105
  logger.info(f"✅ Bucket '{BUCKET_ID}' is ready (alternative format)")
106
  except Exception as e2: